update setup

545999961 · 545999961 · commit e2ca52e1d0d2 · 2024-10-30T23:14:54.000+08:00
diff --git a/FlagEmbedding/evaluation/mteb/searcher.py b/FlagEmbedding/evaluation/mteb/searcher.py
@@ -39,7 +39,7 @@ def encode(self, corpus: List[Dict[str, str]], **kwargs):
             input_texts = ['{} {}'.format(doc.get('title', ''), doc['text']).strip() for doc in corpus]
         else:
             input_texts = corpus
-        emb = self.embedder.encode_corpus(input_texts)
+        emb = self.embedder.encode_queries(input_texts)
         if isinstance(emb, dict):
             emb = emb["dense_vecs"]
         return emb
diff --git a/imgs/bge_logo.jpg b/imgs/bge_logo.jpg
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
     include_package_data=True,
     install_requires=[
         'torch>=1.6.0',
-        'transformers==4.45.2',
+        'transformers==4.44.2',
         'datasets==2.19.0',
         'accelerate>=0.20.1',
         'sentence_transformers',
