[data] fix template (#8827)

2025-08-06 06:58:09 +08:00
parent 4dfad24902
commit b523543994
4 changed files with 8 additions and 4 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -1069,7 +1069,9 @@ register_template(
    format_assistant=StringFormatter(slots=["{{content}}<|end|>"]),
    format_system=StringFormatter(slots=["<|start|>system<|message|>{{content}}<|end|>"]),
    default_system="You are ChatGPT, a large language model trained by OpenAI.",
+    thought_words=("<|channel|>analysis<|message|>", "<|end|><|start|>assistant<|channel|>final<|message|>"),
    efficient_eos=True,
+    template_class=ReasoningTemplate,
 )