mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-08-04 03:10:21 +00:00
added progressbar
added an option to disable progressbar added interrupt support to DDIM/PLMS
This commit is contained in:
@@ -153,6 +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
|
||||
|
||||
for n in range(p.n_iter):
|
||||
if state.interrupted:
|
||||
break
|
||||
@@ -207,6 +209,8 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
|
||||
|
||||
output_images.append(image)
|
||||
|
||||
state.nextjob()
|
||||
|
||||
unwanted_grid_because_of_img_count = len(output_images) < 2 and opts.grid_only_if_multiple
|
||||
if not p.do_not_save_grid and not unwanted_grid_because_of_img_count:
|
||||
return_grid = opts.return_grid
|
||||
|
Reference in New Issue
Block a user