1inkusFace commited on
Commit
9d2f41d
·
verified ·
1 Parent(s): 97f63b2

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +2 -2
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1206,8 +1206,8 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1206
  # FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
1207
 
1208
  # 1. Stack the image embeddings
1209
- stacked_image_embeds = torch.stack(image_prompt_embeds_list)
1210
- stacked_image_embedsg = torch.cat(image_prompt_embeds_list, dim=1)
1211
  print('shape 1: ', stacked_image_embeds.shape)
1212
  print('shape 1a: ', stacked_image_embedsg.shape)
1213
  # 2. Calculate the mean of the stacked embeddings
 
1206
  # FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
1207
 
1208
  # 1. Stack the image embeddings
1209
+ stacked_image_embedsg = torch.stack(image_prompt_embeds_list)
1210
+ stacked_image_embeds = torch.cat(image_prompt_embeds_list, dim=1)
1211
  print('shape 1: ', stacked_image_embeds.shape)
1212
  print('shape 1a: ', stacked_image_embedsg.shape)
1213
  # 2. Calculate the mean of the stacked embeddings