mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-12 20:15:26 +00:00
Add sentence-transformers library to requirements.txt and comment out model_name in base.py
This commit is contained in:
1 parent
318de530e3
commit
a2418d1373
2 files changed
+2
-1
No files matched your search
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user