We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 376742a commit 485ae86Copy full SHA for 485ae86
redisvl/utils/vectorize/text/huggingface.py
@@ -99,7 +99,7 @@ def embed(
99
100
if preprocess:
101
text = preprocess(text)
102
- embedding = self._client.encode([text])[0]
+ embedding = self._client.encode([text], **kwargs)[0]
103
return self._process_embedding(embedding.tolist(), as_buffer, **kwargs)
104
105
def embed_many(
@@ -135,7 +135,7 @@ def embed_many(
135
136
embeddings: List = []
137
for batch in self.batchify(texts, batch_size, preprocess):
138
- batch_embeddings = self._client.encode(batch)
+ batch_embeddings = self._client.encode(batch, **kwargs)
139
embeddings.extend(
140
[
141
self._process_embedding(embedding.tolist(), as_buffer, **kwargs)
0 commit comments