fix: specify embedding retriever version (#7811)

deepset-ai · Jun 6, 2024 · 1c5fa02 · 1c5fa02
1 parent 7d038c8
commit 1c5fa02
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/haystack/nodes/retriever/_embedding_encoder.py b/haystack/nodes/retriever/_embedding_encoder.py
@@ -135,7 +135,10 @@ def __init__(self, retriever: "EmbeddingRetriever"):
         # e.g. 'roberta-base-nli-stsb-mean-tokens'
         torch_and_transformers_import.check()
         self.embedding_model = SentenceTransformer(
-            retriever.embedding_model, device=str(retriever.devices[0]), use_auth_token=retriever.use_auth_token
+            retriever.embedding_model,
+            device=str(retriever.devices[0]),
+            use_auth_token=retriever.use_auth_token,
+            revision=retriever.model_version,
         )
         self.batch_size = retriever.batch_size
         self.embedding_model.max_seq_length = retriever.max_seq_len

diff --git a/pyproject.toml b/pyproject.toml
@@ -87,7 +87,7 @@ dependencies = [
 [project.optional-dependencies]
 inference = [
   "transformers[torch,sentencepiece]==4.39.3",
-  "sentence-transformers>=2.2.0",  # See haystack/nodes/retriever/_embedding_encoder.py, _SentenceTransformersEmbeddingEncoder
+  "sentence-transformers>=2.3.1",  # See haystack/nodes/retriever/_embedding_encoder.py, _SentenceTransformersEmbeddingEncoder
   "huggingface-hub>=0.5.0",
 ]
 elasticsearch = [

diff --git a/releasenotes/notes/specify-sentence-transformer-model-version-55b633e8b294189c.yaml b/releasenotes/notes/specify-sentence-transformer-model-version-55b633e8b294189c.yaml
@@ -0,0 +1,5 @@
+---
+fixes:
+  - |
+    Fix bug causing latest version of sentence transformer model always being downloaded, even if specific version
+    is given.