Add sentence-transformers library to requirements.txt and comment out model_name in base.py

This commit is contained in:
Alex
2024-01-29 20:51:28 +00:00
parent 318de530e3
commit a2418d1373
2 changed files with 2 additions and 1 deletions

View File

@@ -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

View File

@@ -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: