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

fix phi-small template

Browse Source 
Former-commit-id: 9c1bbfac2173450c56ded2cf343e3a37ba369b7b
This commit is contained in:
menibrief 2024-09-18 23:52:30 +03:00 committed by GitHub
parent c9cd388630
commit 4ed2b629a5
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -821,6 +821,15 @@ _register_template(
replace_eos=True,
)
_register_template(
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"]),
format_prefix=EmptyFormatter(slots=[{"<|endoftext|>"}]),
stop_words=["<|end|>"],
replace_eos=True,
)
_register_template(
name="qwen",