1inkusFace commited on
Commit
efadbd8
·
verified ·
1 Parent(s): 2213cfd

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +1 -0
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1177,6 +1177,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1177
  # Concatenate the image embeddings
1178
  ## clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0)
1179
  clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0).unsqueeze(0)
 
1180
  clip_image_embeds = clip_image_embeds.view(bs_embed, seq_len, -1) # Simplified reshape
1181
 
1182
  # 4. Prepare timesteps
 
1177
  # Concatenate the image embeddings
1178
  ## clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0)
1179
  clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0).unsqueeze(0)
1180
+ bs_embed, seq_len, _ = clip_image_embeds.shape
1181
  clip_image_embeds = clip_image_embeds.view(bs_embed, seq_len, -1) # Simplified reshape
1182
 
1183
  # 4. Prepare timesteps