From 395f1705979193a0525e1479f0d263e9751e3793 Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Tue, 6 Sep 2022 10:11:25 +0300 Subject: readme extras for VRAM for added missing packages to requirements for #74 add support for negative numbers in X/Y plot (plus ranges) #73 changed progressbar to work properly with custom modes --- modules/processing.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'modules/processing.py') diff --git a/modules/processing.py b/modules/processing.py index 1351579b..e8923a7a 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -153,7 +153,8 @@ def process_images(p: StableDiffusionProcessing) -> Processed: with torch.no_grad(), precision_scope("cuda"), ema_scope(): p.init() - state.job_count = p.n_iter + if state.job_count == -1: + state.job_count = p.n_iter for n in range(p.n_iter): if state.interrupted: -- cgit v1.2.1