fix #2147

Former-commit-id: 49445a03cd46af4e7036cf444cd041dfab2d8941
2025-10-16 00:28:10 +08:00 · 2024-01-12 03:30:56 +08:00 · 2024-01-12 03:30:56 +08:00 · c5f099138d
commit c5f099138d
parent 6e64e02f71
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=[]