Unverified Commit 3ad49eee authored by Antoine Bouthors's avatar Antoine Bouthors Committed by GitHub
Browse files

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
parent 769f0be8
...@@ -632,6 +632,9 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline): ...@@ -632,6 +632,9 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
# 4. Preprocess mask and image # 4. Preprocess mask and image
if isinstance(image, PIL.Image.Image) and isinstance(mask_image, PIL.Image.Image): if isinstance(image, PIL.Image.Image) and isinstance(mask_image, PIL.Image.Image):
mask, masked_image = prepare_mask_and_masked_image(image, mask_image) mask, masked_image = prepare_mask_and_masked_image(image, mask_image)
else:
mask = mask_image
masked_image = image * (mask < 0.5)
# 5. set timesteps # 5. set timesteps
self.scheduler.set_timesteps(num_inference_steps, device=device) self.scheduler.set_timesteps(num_inference_steps, device=device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment