diff --git a/milvus_model/hybrid/bge_m3.py b/milvus_model/hybrid/bge_m3.py index 22f8d24..2a14b6c 100644 --- 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 index 5e5aeb6..fdd6a82 100644 --- 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")