Merge pull request #7319 from Thurion/img2img_batch_fix

Fix error when using img2img batch without masks
2023-01-28 17:31:39 +03:00 · 2023-01-28 17:31:39 +03:00 · cd7e8fb42b
parent b7d2af8c7f 1421e95960
commit cd7e8fb42b
1 changed files with 4 additions and 2 deletions
--- a/modules/img2img.py
+++ b/modules/img2img.py
@ -21,8 +21,10 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):

    images = shared.listfiles(input_dir)

+    is_inpaint_batch = False
+    if inpaint_mask_dir:
        inpaint_masks = shared.listfiles(inpaint_mask_dir)
-    is_inpaint_batch = inpaint_mask_dir and len(inpaint_masks) > 0
+        is_inpaint_batch = len(inpaint_masks) > 0
    if is_inpaint_batch:
        print(f"\nInpaint batch is enabled. {len(inpaint_masks)} masks found.")