fix: Custom Models do not receive prompt_type or task_name (#1491)

embeddings-benchmark · Nov 23, 2024 · d3569a8 · d3569a8
1 parent 3ff38ec
commit d3569a8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mteb/models/sentence_transformer_wrapper.py b/mteb/models/sentence_transformer_wrapper.py
@@ -98,7 +98,9 @@ def encode(
 
         embeddings = self.model.encode(
             sentences,
+            task_name=task_name,
             prompt_name=prompt_name,
+            prompt_type=prompt_type,
             **kwargs,
         )
         if isinstance(embeddings, torch.Tensor):