making things work-ish (#560)

* making things work-ish * BLACK!!!!
mlcommons · Oct 7, 2024 · 8aeca34 · 8aeca34
1 parent 0884d7c
commit 8aeca34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/modelgauge/private_ensemble_annotator_set.py b/src/modelgauge/private_ensemble_annotator_set.py
@@ -14,7 +14,7 @@
         EnsembleAnnotator,
     )  # type: ignore
     from modelgauge.annotators.llama_3_70b_chat_annotator import (
-        LLAMA_3_70B_CONFIG,
+        LLAMA_3_70B_PE_TAMALPAIS_2024_09_09_CONFIG as LLAMA_3_70B_CONFIG,
     )  # type: ignore
     from modelgauge.annotators.llama_guard_2_lora_annotator import (
         LlamaGuard2LoRAAnnotator,
@@ -23,7 +23,7 @@
         config as LG2_LORA_CONFIG,
     )  # type: ignore
     from modelgauge.annotators.mistral_8x22b_instruct_annotator import (
-        MISTRAL_8x22B_CONFIG,
+        MISTRAL_8x22B_PE_TAMALPAIS_2024_09_09_CONFIG as MISTRAL_8x22B_CONFIG,
     )  # type: ignore
     from modelgauge.annotators.prompt_engineered_annotator import (
         PromptEngineeredAnnotator,