From 9f8a721313a4e28e8ce3d80b116d2ca564c56017 Mon Sep 17 00:00:00 2001 From: Hubert <42952108+yingfhu@users.noreply.github.com> Date: Tue, 7 Nov 2023 20:01:23 +0800 Subject: [PATCH] [Fix] fix registry error with internal (#551) * [Fix] fix conflict with internal * [Fix] fix conflict with internal --- opencompass/models/minimax_api.py | 2 -- opencompass/models/xunfei_api.py | 2 -- opencompass/models/zhipuai_api.py | 2 -- 3 files changed, 6 deletions(-) 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.