From b99a47b12d98c7324ae174c2a3418d999161f1bd Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sun, 4 Sep 2022 03:38:24 +0300
Subject: save grids created by scripts

---
 scripts/prompt_matrix.py | 3 +++
 scripts/xy_grid.py       | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

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
-- 
cgit v1.2.1