From 9f7afedabbb954c868759abd20336077729b2959 Mon Sep 17 00:00:00 2001
From: Norman Mu <normster@users.noreply.github.com>
Date: Thu, 12 Oct 2023 14:54:21 -0700
Subject: [PATCH] Fix Mistral template (#2529)

---
 fastchat/conversation.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/fastchat/conversation.py b/fastchat/conversation.py
index 032aea9ea..e7863f03b 100644
--- 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>",
     )
 )