Update prompt_matrix.py

1
2023-02-23 01:55:07 +08:00 · 2023-02-23 01:55:07 +08:00 · 2fa91cbee6
parent 2c58d373dd
commit 2fa91cbee6
1 changed files with 1 additions and 1 deletions
--- a/scripts/prompt_matrix.py
+++ b/scripts/prompt_matrix.py
@ -100,7 +100,7 @@ class Script(scripts.Script):
        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.hr_upscale_to_x, p.hr_upscale_to_y,, prompt_matrix_parts, margin_size)
+        grid = images.draw_prompt_matrix(grid, processed.images[0].width, processed.images[0].height, prompt_matrix_parts, margin_size)
        processed.images.insert(0, grid)
        processed.index_of_first_image = 1
        processed.infotexts.insert(0, processed.infotexts[0])