diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-29 09:29:23 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-29 09:29:23 +0300 |
| commit | 1d24665229bcef3de1e517f0db3ce296749d8a98 (patch) | |
| tree | f733b81f163515e2c7fc1c28250711c733ae12e7 /modules | |
| parent | 0a8515085ef258d4b76fdc000f7ed9d55751d6b8 (diff) | |
| parent | 09a142a05a6da8bdd4f36678a098c2a573db181a (diff) | |
Merge pull request #7344 from glop102/master
Reduce grid rows if larger than number of images available
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/images.py | 2 |
1 files changed, 2 insertions, 0 deletions
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)
|