diff --git a/opencompass/models/__init__.py b/opencompass/models/__init__.py index fedaf904..92f0ce1a 100644 --- a/opencompass/models/__init__.py +++ b/opencompass/models/__init__.py @@ -17,7 +17,7 @@ from .minimax_api import MiniMax # noqa: F401 from .mixtral import Mixtral # noqa: F401 from .modelscope import ModelScope, ModelScopeCausalLM # noqa: F401, F403 from .moonshot_api import MoonShot # noqa: F401 -from .openai_api import OpenAI, OpenAIAllesAPIN # noqa: F401 +from .openai_api import OpenAI # noqa: F401 from .pangu_api import PanGu # noqa: F401 from .sensetime_api import SenseTime # noqa: F401 from .turbomind import TurboMindModel # noqa: F401 diff --git a/opencompass/models/openai_api.py b/opencompass/models/openai_api.py index e71ee096..3228ec2f 100644 --- a/opencompass/models/openai_api.py +++ b/opencompass/models/openai_api.py @@ -312,7 +312,6 @@ class OpenAI(BaseAPIModel): return prompt -@MODELS.register_module(name=['OpenAIAllesAPIN']) class OpenAIAllesAPIN(OpenAI): """Model wrapper around OpenAI-AllesAPIN.