1inkusFace commited on
Commit
b76a5d9
·
verified ·
1 Parent(s): 9032eab

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +3 -3
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1188,9 +1188,9 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1188
  # FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
1189
 
1190
  clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0).unsqueeze(0)
1191
- bs_embed, seq_len, _ = image_prompt_embeds.shape
1192
- image_prompt_embeds = image_prompt_embeds.repeat(1, 1, 1)
1193
- image_prompt_embeds = image_prompt_embeds.view(bs_embed * 1, seq_len, -1)
1194
 
1195
  #clip_image_embeds = torch.cat([torch.stack(image_prompt_embeds_list)], dim=0).mean(dim=0)
1196
 
 
1188
  # FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
1189
 
1190
  clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0).unsqueeze(0)
1191
+ bs_embed, seq_len, _ = clip_image_embeds.shape
1192
+ clip_image_embeds = clip_image_embeds.repeat(1, 1, 1)
1193
+ clip_image_embeds = clip_image_embeds.view(bs_embed * 1, seq_len, -1)
1194
 
1195
  #clip_image_embeds = torch.cat([torch.stack(image_prompt_embeds_list)], dim=0).mean(dim=0)
1196