mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implemented batch count logic to Outpainting mk2

Browse Source
This commit is contained in:
wywywywy 2022-10-20 15:31:09 +01:00 committed by GitHub
parent 7f8ab1ee8f
commit 4281f255d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
scripts/outpainting_mk_2.py
View File

@ -242,7 +242,14 @@ class Script(scripts.Script):
out = out.crop((0, 0, res_w, res_h)) out = out.crop((0, 0, res_w, res_h))
return out return out
batch_count = p.n_iter
p.n_iter = 1
state.job_count = batch_count
all_images = []
for i in range(batch_count):
img = init_image img = init_image
state.job = f"Batch {i + 1} out of {state.job_count}"
if left > 0: if left > 0:
img = expand(img, left, is_left=True) img = expand(img, left, is_left=True)
@ -253,10 +260,19 @@ class Script(scripts.Script):
if down > 0: if down > 0:
img = expand(img, down, is_bottom=True) img = expand(img, down, is_bottom=True)
res = Processed(p, [img], initial_seed_and_info[0], initial_seed_and_info[1]) all_images.append(img)
combined_grid_image = images.image_grid(all_images)
if opts.return_grid:
all_images = [combined_grid_image] + all_images
res = Processed(p, all_images, initial_seed_and_info[0], initial_seed_and_info[1])
if opts.samples_save: if opts.samples_save:
images.save_image(img, p.outpath_samples, "", res.seed, p.prompt, opts.grid_format, info=res.info, p=p) images.save_image(img, p.outpath_samples, "", res.seed, p.prompt, opts.grid_format, info=res.info, p=p)
if opts.grid_save:
images.save_image(combined_grid_image, p.outpath_grids, "grid", res.seed, p.prompt, opts.grid_format, info=res.info, short_filename=not opts.grid_extended_filename, grid=True, p=p)
return res return res