apply instruction template

embeddings-benchmark · Jan 7, 2025 · ddf96f4 · ddf96f4
1 parent 27df66a
commit ddf96f4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mteb/models/instruct_wrapper.py b/mteb/models/instruct_wrapper.py
@@ -121,6 +121,8 @@ def encode(
     ) -> np.ndarray:
         task = mteb.get_task(task_name=task_name)
         instruction = self.get_instruction(task_name, prompt_type)
+        if self.instruction_template:
+            instruction = self.format_instruction(instruction, prompt_type)
 
         # to passage prompts won't be applied to passages
         if (