@@ -631,10 +631,22 @@ def register_model_group(
631
631
632
632
register_model_group (
633
633
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
+ },
634
638
"InternLM2.5-7B" : {
635
639
DownloadSource .DEFAULT : "internlm/internlm2_5-7b" ,
636
640
DownloadSource .MODELSCOPE : "Shanghai_AI_Laboratory/internlm2_5-7b" ,
637
641
},
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
+ },
638
650
"InternLM2.5-7B-Chat" : {
639
651
DownloadSource .DEFAULT : "internlm/internlm2_5-7b-chat" ,
640
652
DownloadSource .MODELSCOPE : "Shanghai_AI_Laboratory/internlm2_5-7b-chat" ,
@@ -643,6 +655,10 @@ def register_model_group(
643
655
DownloadSource .DEFAULT : "internlm/internlm2_5-7b-chat-1m" ,
644
656
DownloadSource .MODELSCOPE : "Shanghai_AI_Laboratory/internlm2_5-7b-chat-1m" ,
645
657
},
658
+ "InternLM2.5-20B-Chat" : {
659
+ DownloadSource .DEFAULT : "internlm/internlm2_5-20b-chat" ,
660
+ DownloadSource .MODELSCOPE : "Shanghai_AI_Laboratory/internlm2_5-20b-chat" ,
661
+ },
646
662
},
647
663
template = "intern2" ,
648
664
)
0 commit comments