From 8c2b7aa1ab4228662cbfbbfca5ae1579bfb34240 Mon Sep 17 00:00:00 2001
From: hoshi-hiyouga <hiyouga@buaa.edu.cn>
Date: Sat, 2 Nov 2024 21:21:22 +0800
Subject: [PATCH] update template

Former-commit-id: 478cbb1aa72f218df37b5a4686db2248ad2605dd
---
 src/llamafactory/data/template.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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]"]),