Skip to content

Commit d00367c

Browse files
committed
Add HFOptionSelectingInferenceEngine get_engine_id method
Signed-off-by: Martín Santillán Cooper <[email protected]>
1 parent 72857db commit d00367c

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/unitxt/inference.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -3293,14 +3293,18 @@ class HFOptionSelectingInferenceEngine(InferenceEngine, TorchDeviceMixin):
32933293
32943294
This class uses models from the HuggingFace Transformers library to calculate log probabilities for text inputs.
32953295
"""
3296-
3296+
label = "hf_option_selection"
32973297
model_name: str
32983298
batch_size: int
32993299

33003300
_requirements_list = {
33013301
"transformers": "Install huggingface package using 'pip install --upgrade transformers"
33023302
}
33033303

3304+
def get_engine_id(self):
3305+
return get_model_and_label_id(self.model, self.label)
3306+
3307+
33043308
def prepare_engine(self):
33053309
from transformers import AutoModelForCausalLM, AutoTokenizer
33063310

0 commit comments

Comments
 (0)