[model] support hunyuan-mt model (#9284)
Co-authored-by: wyfdgg <liwenkun0812@163.com> Co-authored-by: Yaowei Zheng <hiyouga@buaa.edu.cn>
This commit is contained in:
@@ -1201,10 +1201,10 @@ register_template(
|
||||
|
||||
register_template(
|
||||
name="hunyuan",
|
||||
format_user=StringFormatter(slots=["<|bos|>user\n{{content}}<|eos|>\n<|bos|>assistant\n"]),
|
||||
format_assistant=StringFormatter(slots=["{{content}}<|eos|>\n"]),
|
||||
format_system=StringFormatter(slots=["<|bos|>system\n{{content}}<|eos|>\n"]),
|
||||
format_prefix=EmptyFormatter(slots=["<|bos|>"]),
|
||||
format_user=StringFormatter(slots=["{{content}}<|extra_0|>"]),
|
||||
format_assistant=StringFormatter(slots=["{{content}}<|eos|>"]),
|
||||
format_system=StringFormatter(slots=["{{content}}<|extra_4|>"]),
|
||||
format_prefix=EmptyFormatter(slots=["<|startoftext|>"]),
|
||||
stop_words=["<|eos|>"],
|
||||
)
|
||||
|
||||
|
||||
@@ -1152,6 +1152,10 @@ register_model_group(
|
||||
DownloadSource.DEFAULT: "tencent/Hunyuan-7B-Instruct",
|
||||
DownloadSource.MODELSCOPE: "AI-ModelScope/Hunyuan-7B-Instruct",
|
||||
},
|
||||
"Hunyuan-MT-7B-Instruct": {
|
||||
DownloadSource.DEFAULT: "tencent/Hunyuan-MT-7B",
|
||||
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/Hunyuan-MT-7B",
|
||||
},
|
||||
},
|
||||
template="hunyuan",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user