Model: Support Qwen 2.5 vl (#3258)
This commit is contained in:
@@ -427,6 +427,8 @@ def match_chat_ml(model_path: str):
|
||||
if "tinyllama" in model_path:
|
||||
return get_chat_template("chatml")
|
||||
# Now the suffix for qwen2 chat model is "instruct"
|
||||
if "qwen" in model_path and "vl" in model_path:
|
||||
return get_chat_template("qwen2-vl")
|
||||
if "qwen" in model_path:
|
||||
if "vl" in model_path:
|
||||
return get_chat_template("qwen2-vl")
|
||||
@@ -443,6 +445,12 @@ def match_chat_ml(model_path: str):
|
||||
return get_chat_template("chatml-llava")
|
||||
|
||||
|
||||
@register_chat_template_matching_function
|
||||
def match_chat_minicpm(model_path: str):
|
||||
if "minicpm" in model_path:
|
||||
return get_chat_template("minicpmv")
|
||||
|
||||
|
||||
@register_chat_template_matching_function
|
||||
def match_chat_yi(model_path: str):
|
||||
model_path = model_path.lower()
|
||||
|
||||
Reference in New Issue
Block a user