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

fix phi-small template

Browse Source 
Former-commit-id: 48fb6bae6245dc6d5f72ebfc1c2bd9ffacd51b86
This commit is contained in:
menibrief 2024-09-18 23:52:30 +03:00 committed by GitHub
parent f34322afd7
commit a7feab45d5
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",