[Feature] Add Interntrain model support (#1548)

Co-authored-by: x54-729 <[email protected]>
open-compass · Sep 23, 2024 · 3356671 · 3356671
1 parent 24915ae
commit 3356671
Show file tree

Hide file tree

Showing 2 changed files with 420 additions and 0 deletions.
diff --git a/opencompass/models/__init__.py b/opencompass/models/__init__.py
@@ -20,6 +20,7 @@
 from .huggingface_above_v4_33 import HuggingFacewithChatTemplate  # noqa: F401
 from .hunyuan_api import Hunyuan  # noqa: F401
 from .intern_model import InternLM  # noqa: F401
+from .interntrain import InternTrain  # noqa: F401
 from .krgpt_api import KrGPT  # noqa: F401
 from .lightllm_api import LightllmAPI, LightllmChatAPI  # noqa: F401
 from .llama2 import Llama2, Llama2Chat  # noqa: F401