From b7f7906a77dc4982a5877c84603d00782e34caa1 Mon Sep 17 00:00:00 2001 From: Leymore Date: Tue, 14 May 2024 22:54:56 +0800 Subject: [PATCH] update run --- opencompass/utils/run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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