mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-12-29 19:05:05 +08:00
xy_grid - Show blank cells when grid is incomplete
This commit is contained in:
parent
c0b1177a32
commit
175baaaca2
@ -2,6 +2,7 @@ from collections import namedtuple
|
||||
from copy import copy
|
||||
import random
|
||||
|
||||
from PIL import Image
|
||||
import numpy as np
|
||||
|
||||
import modules.scripts as scripts
|
||||
@ -108,7 +109,10 @@ def draw_xy_grid(p, xs, ys, x_labels, y_labels, cell, draw_legend):
|
||||
if first_pocessed is None:
|
||||
first_pocessed = processed
|
||||
|
||||
res.append(processed.images[0])
|
||||
try:
|
||||
res.append(processed.images[0])
|
||||
except:
|
||||
res.append(Image.new(res[0].mode, res[0].size))
|
||||
|
||||
grid = images.image_grid(res, rows=len(ys))
|
||||
if draw_legend:
|
||||
|
Loading…
Reference in New Issue
Block a user