Remove fix_newlines() calls from chat.py

oobabooga · Apr 16, 2023 · 6a03ad0 · 6a03ad0
1 parent 5342f72
commit 6a03ad0
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/modules/chat.py b/modules/chat.py
@@ -13,8 +13,7 @@
 import modules.extensions as extensions_module
 import modules.shared as shared
 from modules.extensions import apply_extensions
-from modules.html_generator import (chat_html_wrapper, fix_newlines,
-                                    make_thumbnail)
+from modules.html_generator import chat_html_wrapper, make_thumbnail
 from modules.text_generation import (encode, generate_reply,
                                      get_max_prompt_length)
 
@@ -60,7 +59,6 @@ def generate_chat_prompt(user_input, state, **kwargs):
     elif not _continue:
 
         # Adding the user message
-        user_input = fix_newlines(user_input)
         if len(user_input) > 0:
             rows.append(f"{prefix1}{user_input}{state['end_of_turn']}\n")
 
@@ -114,7 +112,6 @@ def extract_message_from_reply(reply, state):
                     continue
                 break
 
-    reply = fix_newlines(reply)
     return reply, next_character_found