rework of callback for #6094

1 files changed, 6 insertions, 4 deletions

diff --git a/modules/images.py b/modules/images.py
index 719aaf3b..f84fd485 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -39,12 +39,14 @@ def image_grid(imgs, batch_size=1, rows=None):
 

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

 

+    params = script_callbacks.ImageGridLoopParams(imgs, cols, rows)

+    script_callbacks.image_grid_callback(params)

+

     w, h = imgs[0].size

-    grid = Image.new('RGB', size=(cols * w, rows * h), color='black')

+    grid = Image.new('RGB', size=(params.cols * w, params.rows * h), color='black')

 

-    for i, img in enumerate(imgs):

-        script_callbacks.image_grid_loop_callback(img)

-        grid.paste(img, box=(i % cols * w, i // cols * h))

+    for i, img in enumerate(params.imgs):

+        grid.paste(img, box=(i % params.cols * w, i // params.cols * h))

 

     return grid