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 b909590 commit 7198b83Copy full SHA for 7198b83
integrations/fastembed/src/haystack_integrations/components/rankers/fastembed/colbert_reranker.py
@@ -104,11 +104,10 @@ def warm_up(self):
104
if v is not None:
105
kwargs[k] = v
106
107
- try:
108
- self._encoder = LateInteractionTextEmbedding(**kwargs)
109
- except TypeError:
110
- self._encoder = LateInteractionTextEmbedding(model_name=self.model, threads=self.threads)
111
-
+ # Only include keys that are not None
+ init_kwargs = {k: v for k, v in kwargs.items() if v is not None}
+ self._encoder = LateInteractionTextEmbedding(**init_kwargs)
+
112
gen_q = self._encoder.query_embed(["warmup"])
113
next(gen_q, None)
114
gen_d = self._encoder.embed(["warmup"])
0 commit comments