prevent weird behavior when pressing interrupt just as image finishes

This commit is contained in:
AUTOMATIC 2022-09-23 08:48:19 +03:00
parent 02f58cde8d
commit c82e32652e

View File

@ -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)