Skip to content

Commit 561ae4d

Browse files
committed
fix constants
Former-commit-id: 653fe70
1 parent fb9280a commit 561ae4d

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

src/llamafactory/extras/constants.py

+16
Original file line numberDiff line numberDiff line change
@@ -631,10 +631,22 @@ def register_model_group(
631631

632632
register_model_group(
633633
models={
634+
"InternLM2.5-1.8B": {
635+
DownloadSource.DEFAULT: "internlm/internlm2_5-1_8b",
636+
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/internlm2_5-1_8b",
637+
},
634638
"InternLM2.5-7B": {
635639
DownloadSource.DEFAULT: "internlm/internlm2_5-7b",
636640
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/internlm2_5-7b",
637641
},
642+
"InternLM2.5-20B": {
643+
DownloadSource.DEFAULT: "internlm/internlm2_5-20b",
644+
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/internlm2_5-20b",
645+
},
646+
"InternLM2.5-1.8B-Chat": {
647+
DownloadSource.DEFAULT: "internlm/internlm2_5-1_8b-chat",
648+
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/internlm2_5-1_8b-chat",
649+
},
638650
"InternLM2.5-7B-Chat": {
639651
DownloadSource.DEFAULT: "internlm/internlm2_5-7b-chat",
640652
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/internlm2_5-7b-chat",
@@ -643,6 +655,10 @@ def register_model_group(
643655
DownloadSource.DEFAULT: "internlm/internlm2_5-7b-chat-1m",
644656
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/internlm2_5-7b-chat-1m",
645657
},
658+
"InternLM2.5-20B-Chat": {
659+
DownloadSource.DEFAULT: "internlm/internlm2_5-20b-chat",
660+
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/internlm2_5-20b-chat",
661+
},
646662
},
647663
template="intern2",
648664
)

0 commit comments

Comments
 (0)