ros/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-02-02 08:33:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

[model] fixed&added Hunyuan models (#9750)

Browse Source
This commit is contained in:
Hertz
2026-01-12 01:15:00 +08:00
committed by GitHub
parent a296723697
commit 4d3621e3d3
4 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/template.py
View File

@@ -1139,7 +1139,7 @@ register_template(
register_template(
name="hunyuan1_5",
name="hunyuan_small",
format_user=StringFormatter(slots=["<｜hy_User｜>{{content}}<｜hy_place▁holder▁no▁8｜>"]),
format_assistant=StringFormatter(slots=["{{content}}<｜hy_place▁holder▁no▁2｜>"]),
format_system=StringFormatter(slots=["{{content}}<｜hy_place▁holder▁no▁3｜>"]),

28
src/llamafactory/extras/constants.py
View File

@@ -1144,14 +1144,34 @@ register_model_group(
register_model_group(
models={
"Hunyuan-0.5B-Instruct": {
DownloadSource.DEFAULT: "tencent/Hunyuan-0.5B-Instruct",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/Hunyuan-0.5B-Instruct",
},
"Hunyuan-1.8B-Instruct": {
DownloadSource.DEFAULT: "tencent/Hunyuan-1.8B-Instruct",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/Hunyuan-1.8B-Instruct",
},
"Hunyuan-4B-Instruct": {
DownloadSource.DEFAULT: "tencent/Hunyuan-4B-Instruct",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/Hunyuan-4B-Instruct",
},
"Hunyuan-7B-Instruct": {
DownloadSource.DEFAULT: "tencent/Hunyuan-7B-Instruct",
DownloadSource.MODELSCOPE: "AI-ModelScope/Hunyuan-7B-Instruct",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/Hunyuan-7B-Instruct",
},
"Hunyuan-MT-7B-Instruct": {
DownloadSource.DEFAULT: "tencent/Hunyuan-MT-7B",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/Hunyuan-MT-7B",
},
"HY-MT1.5-7B-Instruct": {
DownloadSource.DEFAULT: "tencent/HY-MT1.5-7B",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/HY-MT1.5-7B",
},
"Hunyuan-A13B-Instruct": {
DownloadSource.DEFAULT: "tencent/Hunyuan-A13B-Instruct",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/Hunyuan-A13B-Instruct",
},
},
template="hunyuan",
)
@@ -1163,12 +1183,8 @@ register_model_group(
DownloadSource.DEFAULT: "tencent/HY-MT1.5-1.8B",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/HY-MT1.5-1.8B",
},
"HY-MT1.5-7B-Instruct": {
DownloadSource.DEFAULT: "tencent/HY-MT1.5-7B",
DownloadSource.MODELSCOPE: "Tencent-Hunyuan/HY-MT1.5-7B",
},
},
template="hunyuan1_5",
template="hunyuan_small",
)