diff --git a/redisvl/utils/vectorize/text/huggingface.py b/redisvl/utils/vectorize/text/huggingface.py index d0a6243d..fdeb0b64 100644 --- a/redisvl/utils/vectorize/text/huggingface.py +++ b/redisvl/utils/vectorize/text/huggingface.py @@ -99,7 +99,7 @@ def embed( if preprocess: text = preprocess(text) - embedding = self._client.encode([text])[0] + embedding = self._client.encode([text], **kwargs)[0] return self._process_embedding(embedding.tolist(), as_buffer, **kwargs) def embed_many( @@ -135,7 +135,7 @@ def embed_many( embeddings: List = [] for batch in self.batchify(texts, batch_size, preprocess): - batch_embeddings = self._client.encode(batch) + batch_embeddings = self._client.encode(batch, **kwargs) embeddings.extend( [ self._process_embedding(embedding.tolist(), as_buffer, **kwargs)