Skip to content

Commit 3ad49ee

Browse files
authored
Fixed mask+masked_image in sd inpaint pipeline (#1516)
* Fixed mask+masked_image in sd inpaint pipeline Those were left unset when inputs are not PIL images * Fixed formatting
1 parent 769f0be commit 3ad49ee

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -632,6 +632,9 @@ def __call__(
632632
# 4. Preprocess mask and image
633633
if isinstance(image, PIL.Image.Image) and isinstance(mask_image, PIL.Image.Image):
634634
mask, masked_image = prepare_mask_and_masked_image(image, mask_image)
635+
else:
636+
mask = mask_image
637+
masked_image = image * (mask < 0.5)
635638

636639
# 5. set timesteps
637640
self.scheduler.set_timesteps(num_inference_steps, device=device)

0 commit comments

Comments
 (0)