From a8bea0a355618a91f9070e5975af1af40323fe7b Mon Sep 17 00:00:00 2001 From: AnonymousCervine Date: Wed, 4 Jan 2023 17:53:38 -0600 Subject: [PATCH] Respect when opts.grid_save and/or opts.samples_save are set to false. --- test_my_prompt/scripts/test_my_prompt_custom_script.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test_my_prompt/scripts/test_my_prompt_custom_script.py b/test_my_prompt/scripts/test_my_prompt_custom_script.py index 70b0d3d..1db98af 100644 --- a/test_my_prompt/scripts/test_my_prompt_custom_script.py +++ b/test_my_prompt/scripts/test_my_prompt_custom_script.py @@ -74,13 +74,14 @@ class Script(scripts.Script): else: proc.images[0] = write_on_image(proc.images[0], "full prompt") - images.save_image(proc.images[0], p.outpath_samples, "", proc.seed, proc.prompt, opts.samples_format, info= proc.info, p=p) + if opts.samples_save: + images.save_image(proc.images[0], p.outpath_samples, "", proc.seed, proc.prompt, opts.samples_format, info= proc.info, p=p) unwanted_grid_because_of_img_count = len(proc.images) < 2 and opts.grid_only_if_multiple if ((opts.return_grid or opts.grid_save) and not p.do_not_save_grid and not unwanted_grid_because_of_img_count) or always_grid: grid = images.image_grid(proc.images) proc.images.insert(0,grid) proc.infotexts.insert(0, proc.infotexts[-1]) - if opts.grid_save or always_grid: + if opts.grid_save: images.save_image(grid, p.outpath_grids, "grid", initial_seed, initial_prompt, opts.grid_format, info=proc.info, short_filename=not opts.grid_extended_filename, p=p, grid=True) return proc