diff options
Diffstat (limited to 'scripts/prompt_matrix.py')
| -rw-r--r-- | scripts/prompt_matrix.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/scripts/prompt_matrix.py b/scripts/prompt_matrix.py index 6340a7d9..b1c486d4 100644 --- a/scripts/prompt_matrix.py +++ b/scripts/prompt_matrix.py @@ -99,8 +99,8 @@ class Script(scripts.Script): p.prompt_for_display = positive_prompt
processed = process_images(p)
- grid = images.image_grid(processed.images, p.batch_size, rows=1 << ((len(prompt_matrix_parts) - 1) // 2))
- grid = images.draw_prompt_matrix(grid, p.width, p.height, prompt_matrix_parts, margin_size)
+ grid = images.image_grid(processed.images, p.batch_size, rows=1 << ((len(prompt_matrix_parts) - 1) // 2))
+ grid = images.draw_prompt_matrix(grid, processed.images[0].width, processed.images[1].height, prompt_matrix_parts, margin_size)
processed.images.insert(0, grid)
processed.index_of_first_image = 1
processed.infotexts.insert(0, processed.infotexts[0])
|