Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add option to prevent empty spots in grid (2/2)

This commit is contained in:
Gugubo 2022-10-14 17:07:24 +02:00 committed by AUTOMATIC1111
parent 43f926aad1
commit 5f87dd1ee0
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/images.py
View File

@ -24,10 +24,13 @@ def image_grid(imgs, batch_size=1, rows=None):
rows = opts.n_rows rows = opts.n_rows
elif opts.n_rows == 0: elif opts.n_rows == 0:
rows = batch_size rows = batch_size
else: elif opts.grid_prevent_empty_spots:
rows = math.floor(math.sqrt(len(imgs))) rows = math.floor(math.sqrt(len(imgs)))
while len(imgs) % rows != 0: while len(imgs) % rows != 0:
rows -= 1 rows -= 1
else:
rows = math.sqrt(len(imgs))
rows = round(rows)
cols = math.ceil(len(imgs) / rows) cols = math.ceil(len(imgs) / rows)