LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 08:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Update Docker build platforms for application and frontend and optimised embedding import

Browse Source
This commit is contained in:
Alex
2024-05-18 12:10:24 +01:00
parent 4534cafd3f
commit 5c8133a810
6 changed files with 14 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
application/vectorstore/base.py
View File

@@ -45,10 +45,15 @@ class BaseVectorStore(ABC):
cohere_api_key=embeddings_key
)
elif embeddings_name == "huggingface_sentence-transformers/all-mpnet-base-v2":
embedding_instance = embeddings_factory[embeddings_name](
#model_name="./model/all-mpnet-base-v2",
model_kwargs={"device": "cpu"},
)
if os.path.exists("./model/all-mpnet-base-v2"):
embedding_instance = embeddings_factory[embeddings_name](
model_name="./model/all-mpnet-base-v2",
model_kwargs={"device": "cpu"},
)
else:
embedding_instance = embeddings_factory[embeddings_name](
model_kwargs={"device": "cpu"},
)
else:
embedding_instance = embeddings_factory[embeddings_name]()