fix zephyr template

yifan1130 · Nov 19, 2023 · 65c6f40 · 65c6f40
1 parent 4a2e9dd
commit 65c6f40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/eval/templates.py b/eval/templates.py
@@ -78,9 +78,9 @@ def create_prompt_with_zephyr_chat_format(messages, bos="<s>", eos="</s>", add_b
 
     for message in messages:
         if message["role"] == "system":
-            formatted_text += "<|system|>\n" + message["content"] + "\n"
+            formatted_text += "<|system|>\n" + message["content"] + eos + "\n"
         elif message["role"] == "user":
-            formatted_text += "<|user|>\n" + message["content"] + "\n"
+            formatted_text += "<|user|>\n" + message["content"] + eos + "\n"
         elif message["role"] == "assistant":
             formatted_text += "<|assistant|>\n" + message["content"] + eos + "\n"
         else: