Merge pull request #14659 from AUTOMATIC1111/immediately-stop-on-seco…

…nd-interrupt immediately stop on second interrupt
AUTOMATIC1111 · Jan 20, 2024 · cfa326b · cfa326b
2 parents 7581da5 + 14b9762
commit cfa326b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/ui_toprow.py b/modules/ui_toprow.py
@@ -107,8 +107,9 @@ def create_submit_box(self):
             )
 
             def interrupt_function():
-                if shared.state.job_count > 1 and shared.opts.interrupt_after_current:
+                if not shared.state.stopping_generation and shared.state.job_count > 1 and shared.opts.interrupt_after_current:
                     shared.state.stop_generating()
+                    gr.Info("Generation will stop after finishing this image, click again to stop immediately.")
                 else:
                     shared.state.interrupt()