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

SD upscale with Batch count>1 results broken images #512

Browse Source
This commit is contained in:
AUTOMATIC 2022-09-15 22:31:06 +03:00
parent 0fde1f3eac
commit 6991f2cb27
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/img2img.py
View File

@ -76,6 +76,7 @@ def img2img(prompt: str, negative_prompt: str, prompt_style: str, prompt_style2:
grid = images.split_grid(img, tile_w=width, tile_h=height, overlap=upscale_overlap) grid = images.split_grid(img, tile_w=width, tile_h=height, overlap=upscale_overlap)
batch_size = p.batch_size
upscale_count = p.n_iter upscale_count = p.n_iter
p.n_iter = 1 p.n_iter = 1
p.do_not_save_grid = True p.do_not_save_grid = True
@ -87,7 +88,7 @@ def img2img(prompt: str, negative_prompt: str, prompt_style: str, prompt_style2:
for tiledata in row: for tiledata in row:
work.append(tiledata[2]) work.append(tiledata[2])
batch_count = math.ceil(len(work) / p.batch_size) batch_count = math.ceil(len(work) / batch_size)
state.job_count = batch_count * upscale_count state.job_count = batch_count * upscale_count
print(f"SD upscaling will process a total of {len(work)} images tiled as {len(grid.tiles[0][2])}x{len(grid.tiles)} per upscale in a total of {state.job_count} batches.") print(f"SD upscaling will process a total of {len(work)} images tiled as {len(grid.tiles[0][2])}x{len(grid.tiles)} per upscale in a total of {state.job_count} batches.")
@ -99,9 +100,10 @@ def img2img(prompt: str, negative_prompt: str, prompt_style: str, prompt_style2:
work_results = [] work_results = []
for i in range(batch_count): for i in range(batch_count):
p.init_images = work[i*p.batch_size:(i+1)*p.batch_size] p.batch_size = batch_size
p.init_images = work[i*batch_size:(i+1)*batch_size]
state.job = f"Batch {i + 1} out of {state.job_count}" state.job = f"Batch {i + 1 + n * batch_count} out of {state.job_count}"
processed = process_images(p) processed = process_images(p)
if initial_info is None: if initial_info is None: