prevent weird behavior when pressing interrupt just as image finishes

2022-09-23 08:48:19 +03:00
parent 02f58cde8d
commit c82e32652e
1 changed files with 1 additions and 0 deletions
@@ -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)