423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 05:32:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

add TeleChat-12B/TeleChat-12B-v2 models

Browse Source 
Former-commit-id: 0dd632fe9e5bbf08605d4b9c6887208b7a127317
This commit is contained in:
hzhaoy 2024-05-29 15:00:37 +08:00
parent 2540ce58c0
commit ce1be3da4b
2 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -918,3 +918,13 @@ _register_template(
format_user=StringFormatter(slots=["<human>:{{content}}\n<bot>:"]),
format_separator=EmptyFormatter(slots=["\n"]),
)
_register_template(
name="telechat",
format_user=StringFormatter(slots=["<_user>{{content}}<_bot>"]),
format_system=StringFormatter(slots=["<_system>{{content}}<_end>"]),
default_system="",
stop_words=["<_end>"],
replace_eos=True,
)

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

@ -1355,3 +1355,19 @@ register_model_group(
},
template="zephyr",
)
register_model_group(
models={
"TeleChat-12B-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat-12B",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat-12B",
},
"TeleChat-12B-v2-Chat": {
DownloadSource.DEFAULT: "Tele-AI/TeleChat-12B-v2",
DownloadSource.MODELSCOPE: "TeleAI/TeleChat-12B-v2",
},
},
module='query,key_value',
template="telechat",
)