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

fix phi template

Browse Source 
Former-commit-id: 14a1ff665eaebfc618229efbe96f09848d52faec
This commit is contained in:
hoshi-hiyouga 2024-04-24 13:55:14 +08:00 committed by GitHub
parent 53d43dea9e
commit 0ef6c579be
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/llmtuner/data/template.py
View File

@ -747,8 +747,10 @@ _register_template(
name="phi", name="phi",
format_user=StringFormatter(slots=["<|user|>\n{{content}}<|end|>\n<|assistant|>\n"]), format_user=StringFormatter(slots=["<|user|>\n{{content}}<|end|>\n<|assistant|>\n"]),
format_system=StringFormatter(slots=[{"bos_token"}, "<|system|>\n{{content}}<|end|>\n"]), format_system=StringFormatter(slots=[{"bos_token"}, "<|system|>\n{{content}}<|end|>\n"]),
format_separator=EmptyFormatter(slots=["<|end|>\n"]), format_separator=EmptyFormatter(slots=["\n"]),
default_system="You are a helpful AI assistant.", default_system="You are a helpful AI assistant.",
stop_words=["<|end|>"],
replace_eos=True,
) )