[ci] fix llama2

RUCAIBox · Jun 11, 2024 · 65b9696 · 65b9696
1 parent a1c6978
commit 65b9696
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/utilization/model/test_apply_prompt_template.py b/tests/utilization/model/test_apply_prompt_template.py
@@ -26,7 +26,7 @@ def test_llama2(conversation: Conversation):
         "This is a system message.\n"
         "<</SYS>>\n"
         "\n"
-        "This is a user message. [/INST] This is an assistant message. </s><s>[INST] This is the second user message. [/INST] This is the second assistant message. </s><s>[INST]"
+        "This is a user message. [/INST] This is an assistant message. </s><s>[INST] This is the second user message. [/INST] This is the second assistant message. </s><s>[INST] "
     )