diff --git a/scripts/prompt_matrix.py b/scripts/prompt_matrix.py index 7087bcde..ed31dd36 100644 --- a/scripts/prompt_matrix.py +++ b/scripts/prompt_matrix.py @@ -79,4 +79,7 @@ class Script(scripts.Script): grid = images.draw_prompt_matrix(grid, p.width, p.height, prompt_matrix_parts) processed.images.insert(0, grid) + if opts.grid_save: + images.save_image(processed.images[0], p.outpath_grids, "prompt_matrix", prompt=original_prompt, seed=seed) + return processed diff --git a/scripts/xy_grid.py b/scripts/xy_grid.py index 2f732687..4a00b4a4 100644 --- a/scripts/xy_grid.py +++ b/scripts/xy_grid.py @@ -150,6 +150,7 @@ class Script(scripts.Script): cell=cell ) - images.save_image(processed.images[0], p.outpath_grids, "xy_grid", prompt=p.prompt, seed=processed.seed) + if opts.grid_save: + images.save_image(processed.images[0], p.outpath_grids, "xy_grid", prompt=p.prompt, seed=processed.seed) return processed