fix examples/community/pipeline_stable_diffusion_xl_instantid.py (#6759)

shaoxiaowang · wangshaoxiao · web-flow · commit e6c9c2513fde · 2024-02-08T10:09:40.000-10:00
Co-authored-by: wangshaoxiao &lt;wangshaoxiao@xiaomi.com&gt;
diff --git a/examples/community/pipeline_stable_diffusion_xl_instantid.py b/examples/community/pipeline_stable_diffusion_xl_instantid.py
@@ -812,6 +812,9 @@ def __call__(
         prompt_image_emb = self._encode_prompt_image_emb(
             image_embeds, device, self.unet.dtype, self.do_classifier_free_guidance
         )
+        bs_embed, seq_len, _ = prompt_image_emb.shape
+        prompt_image_emb = prompt_image_emb.repeat(1, num_images_per_prompt, 1)
+        prompt_image_emb = prompt_image_emb.view(bs_embed * num_images_per_prompt, seq_len, -1)
 
         # 4. Prepare image
         if isinstance(controlnet, ControlNetModel):

Original file line number	Diff line number	Diff line change
`@@ -812,6 +812,9 @@ def __call__(`
`812`	`812`	`prompt_image_emb = self._encode_prompt_image_emb(`
`813`	`813`	`image_embeds, device, self.unet.dtype, self.do_classifier_free_guidance`
`814`	`814`	`)`
	`815`	`+ bs_embed, seq_len, _ = prompt_image_emb.shape`
	`816`	`+ prompt_image_emb = prompt_image_emb.repeat(1, num_images_per_prompt, 1)`
	`817`	`+ prompt_image_emb = prompt_image_emb.view(bs_embed * num_images_per_prompt, seq_len, -1)`
`815`	`818`
`816`	`819`	`# 4. Prepare image`
`817`	`820`	`if isinstance(controlnet, ControlNetModel):`