You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Describe the bug
Sentence transformer models do not allow version to be specified.
Error message
When passing a model_version argument to nodes.EmbeddingRetriever, it ignores this and only downloads the latest version of the sentence-transformer model from huggingface (or from cache).
Expected behavior
Download the specified version.
Additional context
This is due to not propagating the revision argument to SentenceTransformer(), via _SentenceTransformersEmbeddingEncoder
from haystack import nodes
nodes.EmbeddingRetriever(
embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
model_version="39c51f707d95c3c40f68b47f1df9ae52319da356",
model_format="sentence_transformers",
)
Check cache to see what models were downloaded:
ls -al ~/.cache/huggingface/hub/models--sentence-transformers--multi-qa-mpnet-base-dot-v1/snapshots
>> 3af7c6da5b3e1bea796ef6c97fe237538cbe6e7f # latest version, not specified version
Describe the bug
Sentence transformer models do not allow version to be specified.
Error message
When passing a
model_version
argument tonodes.EmbeddingRetriever
, it ignores this and only downloads the latest version of thesentence-transformer
model from huggingface (or from cache).Expected behavior
Download the specified version.
Additional context
This is due to not propagating the
revision
argument toSentenceTransformer()
, via_SentenceTransformersEmbeddingEncoder
Here, we should have passed the version:
To Reproduce
Check cache to see what models were downloaded:
FAQ Check
System:
EmbeddingRetriever
The text was updated successfully, but these errors were encountered: