Update stop sequences for chat

abetlen · Apr 15, 2023 · a6372a7 · a6372a7
1 parent 83b2be6
commit a6372a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
@@ -701,7 +701,7 @@ def create_chat_completion(
             for message in messages
         )
         PROMPT = chat_history + "### Assistant:"
-        PROMPT_STOP = ["### Assistant:", "### Human:", "\n"]
+        PROMPT_STOP = ["### Assistant:", "### Human:"]
         completion_or_chunks = self(
             prompt=PROMPT,
             stop=PROMPT_STOP + stop,