prevent weird behavior when pressing interrupt just as image finishes

3 years ago · c82e32652e
parent 02f58cde8d
commit c82e32652e
1 changed files with 1 additions and 0 deletions
--- 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)