swappy (#1155)

Author: bkorycki

src/modelgauge/private_ensemble_annotator_set.py

@@ -11,7 +11,7 @@
         config as MISTRAL_7B_RUBY_CONFIG,
     )  # type: ignore
     from modelgauge.annotators.mistral_8x22b_instruct_annotator import (
-        MISTRAL_8x22B_PE_TAMALPAIS_2024_09_09_CONFIG as MISTRAL_8x22B_CONFIG,
+        MISTRAL_8x22B_PE_TAMALPAIS_2024_09_09_ON_TOGETHER_DEDICATED as MISTRAL_8x22B_CONFIG,
     )  # type: ignore
     from modelgauge.ensemble.majority_vote_ensemble_strategy import MajorityVoteEnsembleStrategy  # type: ignore
     from modelgauge.safety_model_response import SafetyModelResponse  # type: ignore

src/modelgauge/suts/together_client.py

@@ -481,7 +481,7 @@ def translate_response(self, request: TogetherInferenceRequest, response: Togeth
     SUTS.register(TogetherChatSUT, uid, model_name, InjectSecret(TogetherApiKey))
 
 DEDICATED_CHAT_MODELS = {
-    "mistral-8x22b-instruct-dedicated-together": "mlc_ai_safety_2/mistralai/Mixtral-8x22B-Instruct-v0.1-26b6d754",
+    "mistral-8x22b-instruct-dedicated-together": "mlc_ai_safety_2/mistralai/Mixtral-8x22B-Instruct-v0.1-e1c7d251",
 }
 for uid, model_name in DEDICATED_CHAT_MODELS.items():
     SUTS.register(TogetherDedicatedChatSUT, uid, model_name, InjectSecret(TogetherApiKey))