Skip to content

GitLab

"references/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "250bac897ede97caa7c5cd7e453bdea187cc1c5b"
Unverified Commit bd06dd02 authored by Hu Ye's avatar Hu Ye Committed by GitHub
Browse files

[inpaint pipeline] fix bug for multiple prompts inputs (#959)

parent b2e2d141
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py
View file @ bd06dd02
...@@ -340,8 +340,8 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline): ...@@ -340,8 +340,8 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
masked_image_latents = 0.18215 * masked_image_latents masked_image_latents = 0.18215 * masked_image_latents
# duplicate mask and masked_image_latents for each generation per prompt, using mps friendly method # duplicate mask and masked_image_latents for each generation per prompt, using mps friendly method
mask = mask.repeat(num_images_per_prompt, 1, 1, 1) mask = mask.repeat(batch_size * num_images_per_prompt, 1, 1, 1)
masked_image_latents = masked_image_latents.repeat(num_images_per_prompt, 1, 1, 1) masked_image_latents = masked_image_latents.repeat(batch_size * num_images_per_prompt, 1, 1, 1)
mask = torch.cat([mask] * 2) if do_classifier_free_guidance else mask mask = torch.cat([mask] * 2) if do_classifier_free_guidance else mask
masked_image_latents = ( masked_image_latents = (
......
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