diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-09-01 04:05:58 +0900 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-09-01 04:05:58 +0900 |
| commit | bd9b3d15e8f631f9475d14a5fd07560c177dc2f3 (patch) | |
| tree | 473571407d514fd1426f66ffa7bc48a6f6d24430 /modules/img2img.py | |
| parent | d39440bfb9d3b20338fc23a78e6655b1e2f7c1d5 (diff) | |
fix batch img2img output dir with script
Diffstat (limited to 'modules/img2img.py')
| -rw-r--r-- | modules/img2img.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/modules/img2img.py b/modules/img2img.py index c81c7ab9..0c6d1af5 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -114,11 +114,14 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal else:
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)
+
if proc is None:
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:
p.override_settings['samples_filename_pattern'] = f'{image_path.stem}-[generation_number]'
else:
|