add deekseek r1 (#840)

mlcommons · Feb 4, 2025 · 81d07b3 · 81d07b3
1 parent 3f2aed3
commit 81d07b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/modelgauge/suts/together_client.py b/src/modelgauge/suts/together_client.py
@@ -407,6 +407,7 @@ def translate_response(self, request: TogetherInferenceRequest, response: Togeth
     "Mistral-7B-Instruct-v0.2": "mistralai/Mistral-7B-Instruct-v0.2",
     "Mixtral-8x7B-Instruct-v0.1": "mistralai/Mixtral-8x7B-Instruct-v0.1",
     "mistral-8x22b-instruct": "mistralai/Mixtral-8x22B-Instruct-v0.1",
+    "deepseek-R1": "deepseek-ai/DeepSeek-R1",
     "deepseek-llm-67b-chat": "deepseek-ai/deepseek-llm-67b-chat",
     "StripedHyena-Nous-7B": "togethercomputer/StripedHyena-Nous-7B",
 }