From cd5c32dc197a6f62c7e328a36507eb202ff6b6a5 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 30 Apr 2025 14:54:05 -0700 Subject: [PATCH] UI: Fix max_updates_second not working --- modules/text_generation.py | 2 -- 1 file changed, 2 deletions(-) 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