feat: add smollm support (#8050)

2025-05-26 04:47:54 -07:00
parent 12292e4283
commit 501e7d8a8f
2 changed files with 34 additions and 0 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -1684,6 +1684,21 @@ register_template(
 )


+register_template(
+    name="smollm",
+    format_system=StringFormatter(
+        slots=["<|im_start|>system\n{{content}}<|im_end|>\n"]
+    ),
+    format_user=StringFormatter(
+        slots=["<|im_start|>user\n{{content}}<|im_end|>\n"]
+    ),
+    format_assistant=StringFormatter(
+        slots=["<|im_start|>assistant\n{{content}}<|im_end|>\n"]
+    ),
+    stop_words=["<|im_end|>"],
+)
+
+
 register_template(
    name="solar",
    format_user=StringFormatter(slots=["### User:\n{{content}}\n\n### Assistant:\n"]),