add empty template

xin-li-67 · Apr 15, 2024 · b3ac14f · b3ac14f
1 parent cf52911
commit b3ac14f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py
@@ -581,6 +581,13 @@ def get_template_and_fix_tokenizer(
 )
 
 
+_register_template(
+    name="empty",
+    format_user=StringFormatter(slots=["{{content}}"]),
+    format_assistant=StringFormatter(slots=["{{content}}"]),
+)
+
+
 _register_template(
     name="falcon",
     format_user=StringFormatter(slots=["User: {{content}}\nFalcon:"]),