1inkusFace commited on
Commit
6d9087a
·
verified ·
1 Parent(s): 7ccbb9a

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +1 -1
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1183,7 +1183,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1183
  #clip_image_embeds = clip_image_embeds.view(bs_embed, seq_len) # Simplified reshape
1184
 
1185
  # experimental way
1186
- clip_image_embeds = torch.cat([torch.zeros_like(image_prompt_embeds_list), image_prompt_embeds_list], dim=0).mean(dim=0)
1187
 
1188
 
1189
  # 4. Prepare timesteps
 
1183
  #clip_image_embeds = clip_image_embeds.view(bs_embed, seq_len) # Simplified reshape
1184
 
1185
  # experimental way
1186
+ clip_image_embeds = torch.cat([torch.zeros_like(torch.stack(image_prompt_embeds_list)), torch.stack(image_prompt_embeds_list)], dim=0).mean(dim=0)
1187
 
1188
 
1189
  # 4. Prepare timesteps