Merge pull request #7344 from glop102/master

Reduce grid rows if larger than number of images available
2023-01-29 09:29:23 +03:00 · 2023-01-29 09:29:23 +03:00 · 1d24665229
commit 1d24665229
parent 0a8515085e 09a142a05a
1 changed files with 2 additions and 0 deletions
--- a/modules/images.py
+++ b/modules/images.py
@ -36,6 +36,8 @@ def image_grid(imgs, batch_size=1, rows=None):
        else:
            rows = math.sqrt(len(imgs))
            rows = round(rows)
+    if rows > len(imgs):
+        rows = len(imgs)

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