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

[data] fix template (#7349)

Browse Source
This commit is contained in:
hoshi-hiyouga 2025-03-17 23:45:20 +08:00 committed by GitHub
parent 48a6584fb1
commit 1d2131e5cb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -777,15 +777,17 @@ register_template(
register_template( register_template(
name="default", name="default",
format_user=StringFormatter(slots=["Human: {{content}}\nAssistant:"]), format_user=StringFormatter(slots=["Human: {{content}}", {"eos_token"}, "\nAssistant:"]),
format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}, "\n"]), format_assistant=StringFormatter(slots=["{{content}}", {"eos_token"}, "\n"]),
format_system=StringFormatter(slots=["System: {{content}}\n"]), format_system=StringFormatter(slots=["System: {{content}}", {"eos_token"}, "\n"]),
replace_jinja_template=True,
) )
register_template( register_template(
name="empty", name="empty",
format_assistant=StringFormatter(slots=["{{content}}"]), format_assistant=StringFormatter(slots=["{{content}}"]),
replace_jinja_template=True,
) )
@ -809,6 +811,7 @@ register_template(
name="fewshot", name="fewshot",
format_assistant=StringFormatter(slots=["{{content}}\n\n"]), format_assistant=StringFormatter(slots=["{{content}}\n\n"]),
efficient_eos=True, efficient_eos=True,
replace_jinja_template=True,
) )