mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

allow empty mask dir

Browse Source
This commit is contained in:
Thurion 2023-01-28 14:42:24 +01:00
parent 5d14f282c2
commit 1421e95960
No known key found for this signature in database
GPG Key ID: 588BE89EC6F7BEFE
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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)
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.")