From eda72e756ec6fa984673bcb8ef1f2614f347c472 Mon Sep 17 00:00:00 2001
From: Hubert <42952108+yingfhu@users.noreply.github.com>
Date: Mon, 18 Dec 2023 15:45:31 +0800
Subject: [PATCH] [Fix] minor fix openai (#711)

---
 opencompass/models/__init__.py   | 2 +-
 opencompass/models/openai_api.py | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

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.