Unverified Commit 9f8a7213 authored by Hubert's avatar Hubert Committed by GitHub
Browse files

[Fix] fix registry error with internal (#551)

* [Fix] fix conflict with internal

* [Fix] fix conflict with internal
parent bb2ecf41
......@@ -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.
......
......@@ -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.
......
......@@ -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.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment