Add sentence-transformers library to requirements.txt and comment out…

… model_name in base.py
arc53 · Jan 29, 2024 · a2418d1 · a2418d1
1 parent 318de53
commit a2418d1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/application/requirements.txt b/application/requirements.txt
@@ -24,6 +24,7 @@ python-dotenv==1.0.1
 redis==5.0.1
 Requests==2.31.0
 retry==0.9.2
+sentence-transformers
 tiktoken==0.5.2
 torch==2.1.2
 tqdm==4.66.1

diff --git a/application/vectorstore/base.py b/application/vectorstore/base.py
@@ -46,7 +46,7 @@ def _get_embeddings(self, embeddings_name, embeddings_key=None):
             )
         elif embeddings_name == "huggingface_sentence-transformers/all-mpnet-base-v2":
             embedding_instance = embeddings_factory[embeddings_name](
-                model_name="./model/all-mpnet-base-v2",
+                #model_name="./model/all-mpnet-base-v2",
                 model_kwargs={"device": "cpu"},
             )
         else: