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

This commit is contained in:
Alex committed 2024-05-18 12:10:24 +01:00
1 parent 4534cafd3f
commit 5c8133a810
6 files changed
+14 -20

No files matched your search

+9 -4
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]()