Fix Mistral template (#2529)

lm-sys · Oct 12, 2023 · 9f7afed · 9f7afed
1 parent 7b0ca39
commit 9f7afed
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fastchat/conversation.py b/fastchat/conversation.py
@@ -846,10 +846,10 @@ def get_conv_template(name: str) -> Conversation:
     Conversation(
         name="mistral",
         system_template="",
-        roles=("[INST] ", " [/INST]"),
+        roles=("[INST]", "[/INST]"),
         sep_style=SeparatorStyle.LLAMA2,
-        sep="",
-        sep2=" </s>",
+        sep=" ",
+        sep2="</s>",
     )
 )