Commit c82e3265 authored by AUTOMATIC's avatar AUTOMATIC

prevent weird behavior when pressing interrupt just as image finishes

parent 02f58cde
......@@ -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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment