diff --git a/modules/chat.py b/modules/chat.py index ad373c2..8bc42dd 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -135,7 +135,7 @@ def chatbot_wrapper(text, max_new_tokens, do_sample, temperature, top_p, typical shared.history['internal'][-1] = [text, reply] shared.history['visible'][-1] = [visible_text, visible_reply] - if not substring_found: + if not substring_found and not shared.args.no_stream: yield shared.history['visible'] if next_character_found: break