diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py index 433b244e..89827f69 100644 --- a/src/llamafactory/data/template.py +++ b/src/llamafactory/data/template.py @@ -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]"]),