prevent weird behavior when pressing interrupt just as image finishes

1 files changed, 1 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index a23d65a1..64c9d462 100644
--- a/webui.py
+++ b/webui.py
@@ -53,6 +53,7 @@ def wrap_gradio_gpu_call(func):
         shared.state.current_latent = None

         shared.state.current_image = None

         shared.state.current_image_sampling_step = 0

+        shared.state.interrupted = False

 

         with queue_lock:

             res = func(*args, **kwargs)