From 485ae862cbb86e3dab276727b01e5853e2b1fdf2 Mon Sep 17 00:00:00 2001 From: antonum Date: Fri, 11 Oct 2024 11:52:36 -0400 Subject: [PATCH] Expose show_progress_bar --- redisvl/utils/vectorize/text/huggingface.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)