update run

liushz · May 14, 2024 · b7f7906 · b7f7906
1 parent 0407d27
commit b7f7906
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opencompass/utils/run.py b/opencompass/utils/run.py
@@ -5,8 +5,8 @@
 from mmengine.config import Config
 
 from opencompass.datasets.custom import make_custom_dataset_config
-from opencompass.models import (VLLM, HuggingFaceBaseModel,
-                                HuggingFaceCausalLM,
+from opencompass.models import (VLLM, HuggingFace, HuggingFaceBaseModel,
+                                HuggingFaceCausalLM, HuggingFaceChatGLM3,
                                 HuggingFacewithChatTemplate, TurboMindModel)
 from opencompass.partitioners import NaivePartitioner, NumWorkerPartitioner
 from opencompass.runners import DLCRunner, LocalRunner, SlurmRunner