Skip to content

Commit 313afb7

Browse files
authored
[StableDiffusionInpaintPipeline] fix batch_size for mask and masked latents (huggingface#1279)
fix bs for mask and masked latents
1 parent ebad089 commit 313afb7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -536,7 +536,7 @@ def __call__(
536536
mask, masked_image_latents = self.prepare_mask_latents(
537537
mask,
538538
masked_image,
539-
batch_size,
539+
batch_size * num_images_per_prompt,
540540
height,
541541
width,
542542
text_embeddings.dtype,

0 commit comments

Comments
 (0)