Skip to content

Commit

Permalink
use correct embed parameter
Browse files Browse the repository at this point in the history
  • Loading branch information
latifboubyan authored Nov 28, 2024
1 parent fdccbf6 commit d0780d6
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ def _embed_batch(
range(0, len(texts_to_embed), batch_size), disable=not self.progress_bar, desc="Calculating embeddings"
):
batch = texts_to_embed[i : i + batch_size]
result = self._client.embed(model=self.model, prompt=batch, options=generation_kwargs)
result = self._client.embed(model=self.model, input=batch, options=generation_kwargs)
all_embeddings.extend(result["embeddings"])

meta["model"] = self.model
Expand Down

0 comments on commit d0780d6

Please sign in to comment.