ソースを参照

Minor fix in the extensions implementation

oobabooga 2 年 前
コミット
5ac24b019e
1 ファイル変更3 行追加3 行削除
  1. 3 3
      modules/chat.py

+ 3 - 3
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:
     if visible_text is None:
         visible_text = text
         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:
     if custom_prompt_generator is None:
         prompt = generate_chat_prompt(text, max_new_tokens, name1, name2, context, chat_prompt_size)
         prompt = generate_chat_prompt(text, max_new_tokens, name1, name2, context, chat_prompt_size)