diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-01 15:37:37 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-01 15:37:55 +0000 |
| commit | e672cfb07418a1a3130d3bf21c14a0d3819f81fb (patch) | |
| tree | f1264d7712ad993e1f51c3b83fc2b145f40032b2 /modules/images.py | |
| parent | 6062c85d4d71fa2d1eef864490f1359b24536bbe (diff) | |
| download | stable-diffusion-webui-gfx803-e672cfb07418a1a3130d3bf21c14a0d3819f81fb.tar.gz stable-diffusion-webui-gfx803-e672cfb07418a1a3130d3bf21c14a0d3819f81fb.tar.bz2 stable-diffusion-webui-gfx803-e672cfb07418a1a3130d3bf21c14a0d3819f81fb.zip | |
rework of callback for #6094
Diffstat (limited to 'modules/images.py')
| -rw-r--r-- | modules/images.py | 10 |
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
|