diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-05 23:09:01 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-05 23:09:01 +0000 |
| commit | a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba (patch) | |
| tree | 8da54414ef8918317a6b7eab14f6ada3bed47d0e /webui.py | |
| parent | b6763fb8847df5a5678f37137e7a702569e5c925 (diff) | |
| download | stable-diffusion-webui-gfx803-a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba.tar.gz stable-diffusion-webui-gfx803-a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba.tar.bz2 stable-diffusion-webui-gfx803-a243bc7859b7ab92a28d28c11b0ed5525fa0d6ba.zip | |
added progressbar
added an option to disable progressbar
added interrupt support to DDIM/PLMS
Diffstat (limited to 'webui.py')
| -rw-r--r-- | webui.py | 7 |
1 files changed, 7 insertions, 0 deletions
@@ -53,6 +53,7 @@ def load_model_from_config(config, ckpt, verbose=False): cached_images = {}
+
def run_extras(image, gfpgan_strength, upscaling_resize, extras_upscaler_1, extras_upscaler_2, extras_upscaler_2_visibility):
processing.torch_gc()
@@ -121,10 +122,16 @@ queue_lock = threading.Lock() def wrap_gradio_gpu_call(func):
def f(*args, **kwargs):
+ shared.state.sampling_step = 0
+ shared.state.job_count = 1
+ shared.state.job_no = 0
+
+
with queue_lock:
res = func(*args, **kwargs)
shared.state.job = ""
+ shared.state.job_count = 0
return res
|