Add engine id

elronbandel · elronbandel · commit 5e15eefd06da · 2025-03-17T14:33:28.000+02:00
Signed-off-by: elronbandel &lt;elronbandel@gmail.com&gt;
diff --git a/src/unitxt/inference.py b/src/unitxt/inference.py
@@ -3295,6 +3295,9 @@ class HFOptionSelectingInferenceEngine(InferenceEngine, TorchDeviceMixin):
         "transformers": "Install huggingface package using 'pip install --upgrade transformers"
     }
 
+    def get_engine_id(self):
+        return get_model_and_label_id(self.model_name, "hf_option_selecting")
+
     def prepare_engine(self):
         from transformers import AutoModelForCausalLM, AutoTokenizer
 

Original file line number	Diff line number	Diff line change
`@@ -3295,6 +3295,9 @@ class HFOptionSelectingInferenceEngine(InferenceEngine, TorchDeviceMixin):`
`3295`	`3295`	`"transformers": "Install huggingface package using 'pip install --upgrade transformers"`
`3296`	`3296`	`}`
`3297`	`3297`
	`3298`	`+ def get_engine_id(self):`
	`3299`	`+ return get_model_and_label_id(self.model_name, "hf_option_selecting")`
	`3300`	`+`
`3298`	`3301`	`def prepare_engine(self):`
`3299`	`3302`	`from transformers import AutoModelForCausalLM, AutoTokenizer`
`3300`	`3303`