diff --git a/modules/text_generation.py b/modules/text_generation.py index 2dabc2ba..7ee1225b 100644 --- a/modules/text_generation.py +++ b/modules/text_generation.py @@ -147,7 +147,7 @@ def stop_everything_event(): def generate_reply_wrapper(question, state, eos_token=None, stopping_strings=None): for reply in generate_reply(question, state, eos_token, stopping_strings, is_chat=False): if shared.model_type not in ['HF_seq2seq']: - reply = reply + question + reply = question + reply yield formatted_outputs(reply, shared.model_name)