Merge pull request #4892 from piamo/main

update deepseek template
2024-07-26 11:49:34 +08:00 · 2024-07-26 11:49:34 +08:00 · 8a2846cfe1
parent 9839c6d1f6 44e48e2b82
commit 8a2846cfe1
1 changed files with 1 additions and 0 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@ -578,6 +578,7 @@ _register_template(
 _register_template(
    name="deepseek",
    format_user=StringFormatter(slots=["User: {{content}}\n\nAssistant:"]),
+    format_system=StringFormatter(slots=["{{content}}\n\n"]),
    format_prefix=EmptyFormatter(slots=[{"bos_token"}]),
 )