Add MPNet model and update vector store for Hugging Face embeddings

This commit is contained in:
Alex committed 2024-01-09 11:39:32 +00:00
1 parent 001c450abb
commit b3461b7134
2 files changed
+11

No files matched your search

+5
View File
@@ -44,6 +44,11 @@ class BaseVectorStore(ABC):
embedding_instance = embeddings_factory[embeddings_name](
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"},
)
else:
embedding_instance = embeddings_factory[embeddings_name]()