Update stop sequences for chat

2023-04-15 12:02:48 -04:00 · 2023-04-15 12:02:48 -04:00 · a6372a7ae5
commit a6372a7ae5
parent 83b2be6dc4
1 changed files with 1 additions and 1 deletions
--- a/llama_cpp/llama.py
+++ b/llama_cpp/llama.py
@ -701,7 +701,7 @@ class Llama:
            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,