diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 17:23:40 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-02 17:23:40 +0000 |
| commit | 3ff0de2c594b786ef948a89efb1814c59bb42117 (patch) | |
| tree | bf0841965fdc5293ec5f437bda7cc85bc0fcb330 /modules/sd_samplers.py | |
| parent | 4ec4af6e0b7addeee5221a03f32d117ccdc875d9 (diff) | |
| download | stable-diffusion-webui-gfx803-3ff0de2c594b786ef948a89efb1814c59bb42117.tar.gz stable-diffusion-webui-gfx803-3ff0de2c594b786ef948a89efb1814c59bb42117.tar.bz2 stable-diffusion-webui-gfx803-3ff0de2c594b786ef948a89efb1814c59bb42117.zip | |
added --disable-console-progressbars to disable progressbars in console
disabled printing prompts to console by default, enabled by --enable-console-prompts
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 92522214..9316875a 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -77,7 +77,9 @@ def extended_tdqm(sequence, *args, desc=None, **kwargs): state.sampling_steps = len(sequence)
state.sampling_step = 0
- for x in tqdm.tqdm(sequence, *args, desc=state.job, file=shared.progress_print_out, **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:
break
@@ -207,7 +209,9 @@ def extended_trange(sampler, count, *args, **kwargs): state.sampling_steps = count
state.sampling_step = 0
- for x in tqdm.trange(count, *args, desc=state.job, file=shared.progress_print_out, **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:
break
|