From 09a142a05a6da8bdd4f36678a098c2a573db181a Mon Sep 17 00:00:00 2001
From: glop102 <alexlikeschess@gmail.com>
Date: Sat, 28 Jan 2023 19:25:52 -0500
Subject: Reduce grid rows if larger than number of images available

When a set number of grid rows is specified in settings, then it leads
to situations where an entire row in the grid is empty.
The most noticable example is the processing preview when the row count
is set to 2, where it shows the preview just fine but with a black
rectangle under it.
---
 modules/images.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'modules')

diff --git a/modules/images.py b/modules/images.py
index 0bc3d524..ae3cdaf4 100644
--- 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)
 
-- 
cgit v1.2.1