additional stop character for Mistral

undreamai · May 27, 2024 · 44e5b43 · 44e5b43
1 parent 35dbe9b
commit 44e5b43
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Runtime/LLMChatTemplates.cs b/Runtime/LLMChatTemplates.cs
@@ -310,7 +310,7 @@ public class MistralInstructTemplate : ChatTemplate
 
         public override string[] GetStop(string playerName, string AIName)
         {
-            return AddStopNewlines(new string[] { "[INST]", "[/INST]" });
+            return AddStopNewlines(new string[] { "</s>", "[INST]", "[/INST]" });
         }
     }
 
@@ -331,7 +331,7 @@ public class MistralChatTemplate : MistralInstructTemplate
 
         public override string[] GetStop(string playerName, string AIName)
         {
-            return AddStopNewlines(new string[] { "[INST]", "[/INST]", "###" });
+            return AddStopNewlines(new string[] { "</s>", "[INST]", "[/INST]", "###" });
         }
     }