From 162ee7a5fc899482622b6b14a512576e4393a19a Mon Sep 17 00:00:00 2001 From: Darrshan Seshathrri Sankar <82897100+Darrshan-Sankar@users.noreply.github.com> Date: Wed, 21 Aug 2024 09:35:10 +0530 Subject: [PATCH] Update embedding_functions.py added **kwargs to the init function of the class SentenceTransformerEmbeddingFunction --- chromadb/utils/embedding_functions.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/chromadb/utils/embedding_functions.py b/chromadb/utils/embedding_functions.py index 4a449b4edfc..82b0b19addf 100644 --- a/chromadb/utils/embedding_functions.py +++ b/chromadb/utils/embedding_functions.py @@ -26,6 +26,7 @@ def __init__( model_name: str = "all-MiniLM-L6-v2", device: str = "cpu", normalize_embeddings: bool = False, + **kwargs: Any, ): if model_name not in self.models: try: @@ -34,7 +35,7 @@ def __init__( raise ValueError( "The sentence_transformers python package is not installed. Please install it with `pip install sentence_transformers`" ) - self.models[model_name] = SentenceTransformer(model_name, device=device) + self.models[model_name] = SentenceTransformer(model_name, device=device, **kwargs) self._model = self.models[model_name] self._normalize_embeddings = normalize_embeddings