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

update template

Browse Source 
Former-commit-id: 3559ef6115a831dcd1adf7210995ffd62890cff6
This commit is contained in:
hoshi-hiyouga 2024-11-02 21:21:22 +08:00 committed by GitHub
parent c2ea5fb618
commit c7c558562e
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]"]),