aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_samplers.py
diff options
context:
space:
mode:
authorTrung Ngo <codem01@gmail.com>2022-10-08 10:33:21 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2022-10-08 10:40:39 +0000
commit00117a07efbbe8482add12262a179326541467de (patch)
tree6a92d8bbbaa9b8a7e12eaf5cbb342b24917aac57 /modules/sd_samplers.py
parent786d9f63aaa4515df82eb2cf357ea92f3dae1e29 (diff)
downloadstable-diffusion-webui-gfx803-00117a07efbbe8482add12262a179326541467de.tar.gz
stable-diffusion-webui-gfx803-00117a07efbbe8482add12262a179326541467de.tar.bz2
stable-diffusion-webui-gfx803-00117a07efbbe8482add12262a179326541467de.zip
check specifically for skipped
Diffstat (limited to 'modules/sd_samplers.py')
-rw-r--r--modules/sd_samplers.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py
index df17e93c..13a8b322 100644
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@@ -106,7 +106,7 @@ def extended_tdqm(sequence, *args, desc=None, **kwargs):
seq = sequence if cmd_opts.disable_console_progressbars else tqdm.tqdm(sequence, *args, desc=state.job, file=shared.progress_print_out, **kwargs)
for x in seq:
- if state.interrupted:
+ if state.interrupted or state.skipped:
break
yield x
@@ -254,7 +254,7 @@ def extended_trange(sampler, count, *args, **kwargs):
seq = range(count) if cmd_opts.disable_console_progressbars else tqdm.trange(count, *args, desc=state.job, file=shared.progress_print_out, **kwargs)
for x in seq:
- if state.interrupted:
+ if state.interrupted or state.skipped:
break
if sampler.stop_at is not None and x > sampler.stop_at: