diff --git a/opencompass/models/minimax_api.py b/opencompass/models/minimax_api.py index e6bc7bb3..813d9162 100644 --- a/opencompass/models/minimax_api.py +++ b/opencompass/models/minimax_api.py @@ -4,7 +4,6 @@ from typing import Dict, List, Optional, Union import requests -from opencompass.registry import MODELS from opencompass.utils.prompt import PromptList from .base_api import BaseAPIModel @@ -12,7 +11,6 @@ from .base_api import BaseAPIModel PromptType = Union[PromptList, str] -@MODELS.register_module(name=['MiniMax']) class MiniMax(BaseAPIModel): """Model wrapper around MiniMax. diff --git a/opencompass/models/xunfei_api.py b/opencompass/models/xunfei_api.py index 02698c61..9ff4780d 100644 --- a/opencompass/models/xunfei_api.py +++ b/opencompass/models/xunfei_api.py @@ -3,7 +3,6 @@ import sys from concurrent.futures import ThreadPoolExecutor from typing import Dict, List, Optional, Union -from opencompass.registry import MODELS from opencompass.utils.prompt import PromptList from .base_api import BaseAPIModel @@ -11,7 +10,6 @@ from .base_api import BaseAPIModel PromptType = Union[PromptList, str] -@MODELS.register_module(name=['XunFei']) class XunFei(BaseAPIModel): """Model wrapper around XunFei. diff --git a/opencompass/models/zhipuai_api.py b/opencompass/models/zhipuai_api.py index 2e35d0ce..7360b12f 100644 --- a/opencompass/models/zhipuai_api.py +++ b/opencompass/models/zhipuai_api.py @@ -2,7 +2,6 @@ import sys from concurrent.futures import ThreadPoolExecutor from typing import Dict, List, Optional, Union -from opencompass.registry import MODELS from opencompass.utils.prompt import PromptList from .base_api import BaseAPIModel @@ -10,7 +9,6 @@ from .base_api import BaseAPIModel PromptType = Union[PromptList, str] -@MODELS.register_module() class ZhiPuAI(BaseAPIModel): """Model wrapper around ZhiPuAI.