1inkusFace commited on
Commit
2be201f
·
verified ·
1 Parent(s): c3e9db2

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +2 -1
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1205,7 +1205,8 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1205
  # FAILS clip_image_embeds = torch.cat(torch.stack(image_prompt_embeds_list), dim=0).mean(dim=0)
1206
  # FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
1207
 
1208
-
 
1209
  # 1. Stack the image embeddings
1210
  stacked_image_embeds = torch.stack(image_prompt_embeds_list)
1211
  print('shape 1: ', stacked_image_embeds.shape)
 
1205
  # FAILS clip_image_embeds = torch.cat(torch.stack(image_prompt_embeds_list), dim=0).mean(dim=0)
1206
  # FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
1207
 
1208
+ clip_image_embedsl = self.encode_clip_image_emb(clip_image, device, dtype)
1209
+ print('old method shape: ', clip_image_embedsl.shape)
1210
  # 1. Stack the image embeddings
1211
  stacked_image_embeds = torch.stack(image_prompt_embeds_list)
1212
  print('shape 1: ', stacked_image_embeds.shape)