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

2022-10-14 17:07:24 +02:00 · 2022-10-14 17:07:24 +02:00 · 5f87dd1ee0
parent 43f926aad1
commit 5f87dd1ee0
1 changed files with 4 additions and 1 deletions
--- a/modules/images.py
+++ b/modules/images.py
@ -24,10 +24,13 @@ def image_grid(imgs, batch_size=1, rows=None):
            rows = opts.n_rows
        elif opts.n_rows == 0:
            rows = batch_size
-        else:
+        elif opts.grid_prevent_empty_spots:
            rows = math.floor(math.sqrt(len(imgs)))
            while len(imgs) % rows != 0:
                rows -= 1
+        else:
+            rows = math.sqrt(len(imgs))
+            rows = round(rows)

    cols = math.ceil(len(imgs) / rows)