OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-04-09 06:19:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix batch img2img output dir with script

Browse Source
This commit is contained in:
w-e-w 2023-09-01 04:05:58 +09:00
parent d39440bfb9
commit bd9b3d15e8
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules/img2img.py
View File

@ -114,11 +114,14 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
else: else:
p.override_settings.pop("sd_model_checkpoint", None) p.override_settings.pop("sd_model_checkpoint", None)
if output_dir:
p.outpath_samples = output_dir
p.override_settings['save_to_dirs'] = False
proc = modules.scripts.scripts_img2img.run(p, *args) proc = modules.scripts.scripts_img2img.run(p, *args)
if proc is None: if proc is None:
if output_dir: if output_dir:
p.outpath_samples = output_dir
p.override_settings['save_to_dirs'] = False
if p.n_iter > 1 or p.batch_size > 1: if p.n_iter > 1 or p.batch_size > 1:
p.override_settings['samples_filename_pattern'] = f'{image_path.stem}-[generation_number]' p.override_settings['samples_filename_pattern'] = f'{image_path.stem}-[generation_number]'
else: else: