Update demo.py

demo.py

@@ -14,7 +14,7 @@ DEFAULT_INSTRUCTIONS = f"""The following is a conversation between a highly know
 """
 RETRY_COMMAND = "/retry"
 STOP_STR = f"\n{USER_NAME}:"
-STOP_SUSPECT_LIST = [":", "\n", "User", "\nUser"]
+STOP_SUSPECT_LIST = [":", "\n", "User"]
 
 
 def chat_accordion():
@@ -109,7 +109,7 @@ def chat():
         stream = llm(
             prompt,
             max_new_tokens=1024,
-            stop=[STOP_STR, "<|endoftext|>"],
+            stop=[STOP_STR, "<|endoftext|>", USER_NAME],
             temperature=temperature,
             top_p=top_p,
             stream=True