diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-09-09 05:56:32 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-09 05:56:32 +0000 |
| commit | 329c8ab93216f9e77002e68aab418d63d82cd872 (patch) | |
| tree | 970e55459529e3b74938b4e4d4479037aec26813 /modules/img2img.py | |
| parent | 259768f27fc4da61000610bc81a16f0152b36550 (diff) | |
| parent | 657404b75b2f214a97281afbec1adcb4313d24eb (diff) | |
| download | stable-diffusion-webui-gfx803-329c8ab93216f9e77002e68aab418d63d82cd872.tar.gz stable-diffusion-webui-gfx803-329c8ab93216f9e77002e68aab418d63d82cd872.tar.bz2 stable-diffusion-webui-gfx803-329c8ab93216f9e77002e68aab418d63d82cd872.zip | |
Merge pull request #12926 from AUTOMATIC1111/fix-batch-img2img-output-dir-with-script
fix batch img2img output dir with script
Diffstat (limited to 'modules/img2img.py')
| -rw-r--r-- | modules/img2img.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/modules/img2img.py b/modules/img2img.py index c81c7ab9..c1cae22f 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -114,15 +114,17 @@ 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
+ if p.n_iter > 1 or p.batch_size > 1:
+ p.override_settings['samples_filename_pattern'] = f'{image_path.stem}-[generation_number]'
+ else:
+ p.override_settings['samples_filename_pattern'] = f'{image_path.stem}'
+
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:
- p.override_settings['samples_filename_pattern'] = f'{image_path.stem}'
process_images(p)
|