diff --git a/modules/text_generation.py b/modules/text_generation.py index 4e3d1d7a..8d091868 100644 --- a/modules/text_generation.py +++ b/modules/text_generation.py @@ -100,8 +100,6 @@ def _generate_reply(question, state, stopping_strings=None, is_chat=False, escap last_update = cur_time yield reply - yield reply - if stop_found or (state['max_tokens_second'] > 0 and shared.stop_everything): break