mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[Fix] fix registry error with internal (#551)
* [Fix] fix conflict with internal * [Fix] fix conflict with internal
This commit is contained in:
parent
bb2ecf416e
commit
9f8a721313
@ -4,7 +4,6 @@ from typing import Dict, List, Optional, Union
|
|||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
from opencompass.registry import MODELS
|
|
||||||
from opencompass.utils.prompt import PromptList
|
from opencompass.utils.prompt import PromptList
|
||||||
|
|
||||||
from .base_api import BaseAPIModel
|
from .base_api import BaseAPIModel
|
||||||
@ -12,7 +11,6 @@ from .base_api import BaseAPIModel
|
|||||||
PromptType = Union[PromptList, str]
|
PromptType = Union[PromptList, str]
|
||||||
|
|
||||||
|
|
||||||
@MODELS.register_module(name=['MiniMax'])
|
|
||||||
class MiniMax(BaseAPIModel):
|
class MiniMax(BaseAPIModel):
|
||||||
"""Model wrapper around MiniMax.
|
"""Model wrapper around MiniMax.
|
||||||
|
|
||||||
|
@ -3,7 +3,6 @@ import sys
|
|||||||
from concurrent.futures import ThreadPoolExecutor
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
from typing import Dict, List, Optional, Union
|
from typing import Dict, List, Optional, Union
|
||||||
|
|
||||||
from opencompass.registry import MODELS
|
|
||||||
from opencompass.utils.prompt import PromptList
|
from opencompass.utils.prompt import PromptList
|
||||||
|
|
||||||
from .base_api import BaseAPIModel
|
from .base_api import BaseAPIModel
|
||||||
@ -11,7 +10,6 @@ from .base_api import BaseAPIModel
|
|||||||
PromptType = Union[PromptList, str]
|
PromptType = Union[PromptList, str]
|
||||||
|
|
||||||
|
|
||||||
@MODELS.register_module(name=['XunFei'])
|
|
||||||
class XunFei(BaseAPIModel):
|
class XunFei(BaseAPIModel):
|
||||||
"""Model wrapper around XunFei.
|
"""Model wrapper around XunFei.
|
||||||
|
|
||||||
|
@ -2,7 +2,6 @@ import sys
|
|||||||
from concurrent.futures import ThreadPoolExecutor
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
from typing import Dict, List, Optional, Union
|
from typing import Dict, List, Optional, Union
|
||||||
|
|
||||||
from opencompass.registry import MODELS
|
|
||||||
from opencompass.utils.prompt import PromptList
|
from opencompass.utils.prompt import PromptList
|
||||||
|
|
||||||
from .base_api import BaseAPIModel
|
from .base_api import BaseAPIModel
|
||||||
@ -10,7 +9,6 @@ from .base_api import BaseAPIModel
|
|||||||
PromptType = Union[PromptList, str]
|
PromptType = Union[PromptList, str]
|
||||||
|
|
||||||
|
|
||||||
@MODELS.register_module()
|
|
||||||
class ZhiPuAI(BaseAPIModel):
|
class ZhiPuAI(BaseAPIModel):
|
||||||
"""Model wrapper around ZhiPuAI.
|
"""Model wrapper around ZhiPuAI.
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user