diff --git a/application/requirements.txt b/application/requirements.txt index b970d982..0bb34365 100644 --- 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 index 293d8016..b7146907 100644 --- a/application/vectorstore/base.py +++ b/application/vectorstore/base.py @@ -46,7 +46,7 @@ class BaseVectorStore(ABC): ) 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: