From 5ac24b019ed6a537de47237ee614c24c76bda7fb Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 25 Feb 2023 02:53:18 -0300
Subject: [PATCH] Minor fix in the extensions implementation

---
 modules/chat.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/modules/chat.py b/modules/chat.py
index 0097af0c..2cbc5c8e 100644
--- a/modules/chat.py
+++ b/modules/chat.py
@@ -103,9 +103,9 @@ def chatbot_wrapper(text, max_new_tokens, do_sample, temperature, top_p, typical
 
     if visible_text is None:
         visible_text = text
-        if shared.args.chat:
-            visible_text = visible_text.replace('\n', '<br>')
-        text = apply_extensions(text, "input")
+    if shared.args.chat:
+        visible_text = visible_text.replace('\n', '<br>')
+    text = apply_extensions(text, "input")
 
     if custom_prompt_generator is None:
         prompt = generate_chat_prompt(text, max_new_tokens, name1, name2, context, chat_prompt_size)