From ec482f3dae8d8c5350e4b7688a3027b955e82d9b Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sat, 24 Jun 2023 11:07:11 -0300 Subject: [PATCH] Apply input extensions after yielding *Is typing...* --- modules/chat.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/chat.py b/modules/chat.py index f6b4c32..ac3e52d 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -170,10 +170,11 @@ def chatbot_wrapper(text, history, state, regenerate=False, _continue=False, loa if visible_text is None: visible_text = text - text = apply_extensions('input', text) # *Is typing...* if loading_message: yield {'visible': output['visible'] + [[visible_text, shared.processing_message]], 'internal': output['internal']} + + text = apply_extensions('input', text) else: text, visible_text = output['internal'][-1][0], output['visible'][-1][0] if regenerate: