fix: Update FlagEmbedding dependency>=1.3.3. (#53)

Signed-off-by: wxywb <[email protected]>
milvus-io · Dec 6, 2024 · 2e93f00 · 2e93f00
1 parent 840a2cd
commit 2e93f00
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/milvus_model/hybrid/bge_m3.py b/milvus_model/hybrid/bge_m3.py
@@ -84,7 +84,7 @@ def dim(self) -> Dict:
         }
 
     def _encode(self, texts: List[str]) -> Dict:
-        output = self.model.encode(queries=texts, **self._encode_config)
+        output = self.model.encode(sentences=texts, **self._encode_config)
         results = {}
         if self._encode_config["return_dense"] is True:
             results["dense"] = list(output["dense_vecs"])

diff --git a/milvus_model/utils/__init__.py b/milvus_model/utils/__init__.py
@@ -33,7 +33,7 @@ def import_sentence_transformers():
 
 def import_FlagEmbedding():
     _check_library("peft", package="peft")
-    _check_library("FlagEmbedding", package="FlagEmbedding")
+    _check_library("FlagEmbedding", package="FlagEmbedding>=1.3.3")
 
 def import_nltk():
     _check_library("nltk", package="nltk>=3.9.1")