diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py index efdd44f3..dd355e97 100644 --- a/src/llmtuner/data/template.py +++ b/src/llmtuner/data/template.py @@ -747,8 +747,10 @@ _register_template( name="phi", format_user=StringFormatter(slots=["<|user|>\n{{content}}<|end|>\n<|assistant|>\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.", + stop_words=["<|end|>"], + replace_eos=True, )