From 9f6e34371f2875f8a57f1643d4024f5fccae6ed8 Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Sun, 11 Jan 2026 23:03:59 +0000 Subject: [PATCH] revert: restore state tracking in control/run.py --- modules/control/run.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/control/run.py b/modules/control/run.py index 393cb089c..726d3aa4f 100644 --- a/modules/control/run.py +++ b/modules/control/run.py @@ -679,5 +679,7 @@ def control_run(state: str = '', # pylint: disable=keyword-arg-before-vararg if len(info_txt) > 0: html_txt = html_txt + infotext_to_html(info_txt[0]) if is_generator: + jobid = shared.state.begin('UI') yield (output_images, blended_image, html_txt, output_filename) + shared.state.end(jobid) return (output_images, blended_image, html_txt, output_filename)