[model] Support Intern-S1-mini (#8976)

This commit is contained in:
Haian Huang(深度眸)
2025-08-20 23:52:51 +08:00
committed by GitHub
parent 022a326ca4
commit 1664657d80
5 changed files with 30 additions and 0 deletions

View File

@@ -1201,6 +1201,17 @@ register_template(
)
register_template(
name="intern_s1",
format_user=StringFormatter(slots=["<|im_start|>user\n{{content}}<|im_end|>\n<|im_start|>assistant\n"]),
format_assistant=StringFormatter(slots=["{{content}}<|im_end|>\n"]),
format_system=StringFormatter(slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]),
format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
stop_words=["<|im_end|>"],
mm_plugin=get_mm_plugin(name="intern_vl", image_token="<image>", video_token="<video>"),
)
# copied from qwen template
register_template(
name="keye_vl",

View File

@@ -1264,6 +1264,16 @@ register_model_group(
multimodal=True,
)
register_model_group(
models={
"Intern-S1-mini": {
DownloadSource.DEFAULT: "internlm/Intern-S1-mini",
DownloadSource.MODELSCOPE: "Shanghai_AI_Laboratory/Intern-S1-mini",
}
},
template="intern_s1",
multimodal=True,
)
register_model_group(
models={

View File

@@ -234,6 +234,9 @@ _register_composite_model(
model_type="internvl",
)
_register_composite_model(
model_type="interns1",
)
_register_composite_model(
model_type="Keye",