From bd9b3d15e8f631f9475d14a5fd07560c177dc2f3 Mon Sep 17 00:00:00 2001
From: w-e-w <40751091+w-e-w@users.noreply.github.com>
Date: Fri, 1 Sep 2023 04:05:58 +0900
Subject: [PATCH] fix batch img2img output dir with script

---
 modules/img2img.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/modules/img2img.py b/modules/img2img.py
index c81c7ab9e..0c6d1af57 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: