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

update template

Browse Source 
Former-commit-id: 478cbb1aa72f218df37b5a4686db2248ad2605dd
This commit is contained in:
hoshi-hiyouga 2024-11-02 21:21:22 +08:00 committed by GitHub
parent d99e164cad
commit 8c2b7aa1ab
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -941,8 +941,9 @@ _register_template(
replace_eos=True,
)
_register_template(
name="phi-small",
name="phi_small",
format_user=StringFormatter(slots=["<|user|>\n{{content}}<|end|>\n<|assistant|>\n"]),
format_system=StringFormatter(slots=["<|system|>\n{{content}}<|end|>\n"]),
format_separator=EmptyFormatter(slots=["\n"]),
@ -951,6 +952,7 @@ _register_template(
replace_eos=True,
)
_register_template(
name="pixtral",
format_user=StringFormatter(slots=["[INST] {{content}} [/INST]"]),