From 4dc16df5706a60c9ec08dc96a1e5d1e1073dadc5 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Fri, 12 Jan 2024 03:30:56 +0800
Subject: [PATCH] fix #2147

Former-commit-id: 9aa1a2fc172730eabc4f14bcb08c0d66fcb2d844
---
 src/llmtuner/data/template.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/src/llmtuner/data/template.py b/src/llmtuner/data/template.py
index a53abc41..03241b26 100644
--- 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=[]