423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 03:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

support telechat2 model

Browse Source 
Former-commit-id: 04f19ed0f3
This commit is contained in:
zhaohu xing
2024-12-12 09:40:00 +08:00
committed by hiyouga
parent 002c7d2867
commit cfb4c42ae4
4 changed files with 58 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1957,13 +1957,36 @@ register_model_group(
DownloadSource.DEFAULT: "Tele-AI/TeleChat-12B",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat-12B",
DownloadSource.OPENMIND: "TeleAI/TeleChat-12B-pt",
}
},
template="telechat",
)
register_model_group(
models={
"TeleChat2-3B-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat2-3B",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat2-3B",
},
"TeleChat-12B-v2-Chat": {
"TeleChat2-7B-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat2-7B",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat2-7B",
},
"TeleChat2-12B-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat-12B-v2",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat-12B-v2",
},
"TeleChat2-35B-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat2-35B",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat2-35B",
},
"TeleChat2-115B-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat2-115B",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat2-115B",
},
},
template="telechat",
template="telechat2",
)