fix #2147

Former-commit-id: 9aa1a2fc172730eabc4f14bcb08c0d66fcb2d844
2025-08-04 12:42:51 +08:00 · 2024-01-12 03:30:56 +08:00 · 2024-01-12 03:30:56 +08:00 · 4dc16df570
commit 4dc16df570
parent 69d966eb1f
1 changed files with 2 additions and 7 deletions
--- a/src/llmtuner/data/template.py
+++ b/src/llmtuner/data/template.py
@ -774,15 +774,10 @@ register_template(
 register_template(
    name="zephyr",
    prefix=[
-        {"token": "<|system|>"},
-        "\n{{system}}",
-        {"token": "</s>"}
+        "<|system|>\n{{system}}</s>",
    ],
    prompt=[
-        {"token": "<|user|>"},
-        "\n{{query}}",
-        {"token": "</s>"},
-        {"token": "<|assistant|>"}
+        "<|user|>\n{{query}}</s><|assistant|>"
    ],
    system="You are a friendly chatbot who always responds in the style of a pirate",
    sep=[]