aboutsummaryrefslogtreecommitdiff
path: root/modules
diff options
context:
space:
mode:
authorw-e-w <40751091+w-e-w@users.noreply.github.com>2023-09-06 20:33:43 +0900
committerw-e-w <40751091+w-e-w@users.noreply.github.com>2023-09-06 20:33:43 +0900
commit657404b75b2f214a97281afbec1adcb4313d24eb (patch)
treed95d0491e8ba2b0e74bf969306f7f52ccd1c49e4 /modules
parent35d1c94549cf75e7e312372d90fee0acc2806426 (diff)
use original filename batch img2img with scripts
Diffstat (limited to 'modules')
-rw-r--r--modules/img2img.py9
1 files changed, 4 insertions, 5 deletions
diff --git a/modules/img2img.py b/modules/img2img.py
index 0c6d1af5..c1cae22f 100644
--- a/modules/img2img.py
+++ b/modules/img2img.py
@@ -117,15 +117,14 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
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:
- 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)