Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #7319 from Thurion/img2img_batch_fix

Browse Source 
Fix error when using img2img batch without masks
This commit is contained in:
AUTOMATIC1111 2023-01-28 17:31:39 +03:00 committed by GitHub
commit cd7e8fb42b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/img2img.py
View File

@ -21,8 +21,10 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):
images = shared.listfiles(input_dir) images = shared.listfiles(input_dir)
inpaint_masks = shared.listfiles(inpaint_mask_dir) is_inpaint_batch = False
is_inpaint_batch = inpaint_mask_dir and len(inpaint_masks) > 0 if inpaint_mask_dir:
inpaint_masks = shared.listfiles(inpaint_mask_dir)
is_inpaint_batch = len(inpaint_masks) > 0
if is_inpaint_batch: if is_inpaint_batch:
print(f"\nInpaint batch is enabled. {len(inpaint_masks)} masks found.") print(f"\nInpaint batch is enabled. {len(inpaint_masks)} masks found.")