Support Mimo-VL (#7579)
Signed-off-by: Xinyuan Tong <justinning0323@outlook.com>
This commit is contained in:
@@ -921,6 +921,19 @@ register_conv_template(
|
||||
)
|
||||
)
|
||||
|
||||
register_conv_template(
|
||||
Conversation(
|
||||
name="mimo-vl",
|
||||
system_message="You are MiMo, an AI assistant developed by Xiaomi.",
|
||||
system_template="<|im_start|>system\n{system_message}",
|
||||
roles=("<|im_start|>user", "<|im_start|>assistant"),
|
||||
sep="<|im_end|>\n",
|
||||
sep_style=SeparatorStyle.ADD_NEW_LINE_SINGLE,
|
||||
stop_str=["<|im_end|>"],
|
||||
image_token="<|vision_start|><|image_pad|><|vision_end|>",
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
register_conv_template(
|
||||
Conversation(
|
||||
@@ -1049,3 +1062,9 @@ def match_phi_4_mm(model_path: str):
|
||||
def match_vila(model_path: str):
|
||||
if re.search(r"vila", model_path, re.IGNORECASE):
|
||||
return "chatml"
|
||||
|
||||
|
||||
@register_conv_template_matching_function
|
||||
def match_mimo_vl(model_path: str):
|
||||
if re.search(r"mimo.*vl", model_path, re.IGNORECASE):
|
||||
return "mimo-vl"
|
||||
|
||||
Reference in New Issue
Block a user