diff --git a/opencompass/utils/run.py b/opencompass/utils/run.py index 893bf86f2..9d7c81bb8 100644 --- 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