Add deepseek-v2.5 template

hiyouga · Sep 21, 2024 · ee059c3 · ee059c3
1 parent af8dfb5
commit ee059c3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/llamafactory/data/template.py b/src/llamafactory/data/template.py
@@ -611,6 +611,14 @@ def get_template_and_fix_tokenizer(tokenizer: "PreTrainedTokenizer", data_args:
 )
 
 
+_register_template(
+    name="deepseek_v2.5",
+    format_user=StringFormatter(slots=["<｜User｜>{{content}}<｜Assistant｜>"]),
+    format_system=StringFormatter(slots=["{{content}}"]),
+    format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
+)
+
+
 _register_template(
     name="deepseekcoder",
     format_user=StringFormatter(slots=["### Instruction:\n{{content}}\n### Response:"]),