From decbbe81f5f92bbb4f5613bc78396b7f03eb26a3 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 10 Sep 2022 13:36:16 +0300
Subject: separated options for sample and grid dir saving for #222

---
 modules/images.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'modules/images.py')

diff --git a/modules/images.py b/modules/images.py
index cfe88fa4..d8e74533 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -244,6 +244,9 @@ def sanitize_filename_part(text):
 
 
 def save_image(image, path, basename, seed=None, prompt=None, extension='png', info=None, short_filename=False, no_prompt=False):
+    # would be better to add this as an argument in future, but will do for now
+    is_a_grid = basename != ""
+
     if short_filename or prompt is None or seed is None:
         file_decoration = ""
     elif opts.save_to_dirs:
@@ -257,7 +260,9 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
     else:
         pnginfo = None
 
-    if opts.save_to_dirs and not no_prompt:
+    save_to_dirs = (is_a_grid and opts.grid_save_to_dirs) or (not is_a_grid and opts.save_to_dirs)
+
+    if save_to_dirs and not no_prompt:
         words = re_nonletters.split(prompt or "")
         if len(words[0]) == 0:
             words = ["empty"]
-- 
cgit v1.2.1