gitea-mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-22 07:10:51 -08:00
Code Issues Packages Projects Releases Wiki Activity

separated options for sample and grid dir saving for #222

Browse Source
This commit is contained in:
AUTOMATIC
2022-09-10 13:36:16 +03:00
parent c92f2ff196
commit decbbe81f5
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules/images.py
View File

@@ -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"]