Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-07 13:30:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix mask return type

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-17 09:00:35 -05:00
parent 2afb8d355e
commit 3192c50241
6 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+2 -2
View File
@@ -169,9 +169,9 @@ def process_diffusers(p: StableDiffusionProcessing):
p.ops.append('inpaint')
if p.task_args.get('mask_image', None) is not None: # provided as override by a module
# p.mask = shared.sd_model.mask_processor.blur(p.task_args['mask_image'], blur_factor=p.mask_blur) if p.mask_blur > 0 else p.task_args['mask_image']
p.mask = run_mask(input_image=p.init_images, input_mask=p.task_args['mask_image'], return_type='grayscale')
p.mask = run_mask(input_image=p.init_images, input_mask=p.task_args['mask_image'], return_type='Grayscale')
elif getattr(p, 'image_mask', None) is not None: # standard
p.mask = run_mask(input_image=p.init_images, input_mask=p.image_mask, return_type='grayscale')
p.mask = run_mask(input_image=p.init_images, input_mask=p.image_mask, return_type='Grayscale')
elif getattr(p, 'mask', None) is not None: # backward compatibility
pass
else: # fallback