diff --git a/plugins/huggingface/modelgauge/suts/huggingface_chat_completion.py b/plugins/huggingface/modelgauge/suts/huggingface_chat_completion.py index e991f36bf..f8b9e0949 100644 --- a/plugins/huggingface/modelgauge/suts/huggingface_chat_completion.py +++ b/plugins/huggingface/modelgauge/suts/huggingface_chat_completion.py @@ -176,6 +176,7 @@ def translate_text_prompt(self, prompt: TextPrompt, options: SUTOptions) -> Hugg "athene-v2-chat": "ijf", "aya-expanse-8b": "rfn", "gemma-2-9b-it": "plf", + "gemma-2-9b-it-simpo": "zop", "llama-3-1-tulu-3-8b": "bzk", # check "llama-3-1-tulu-3-70b": "ome", "mistral-nemo-instruct-2407": "mgt",