1inkusFace commited on
Commit
c2041c6
·
verified ·
1 Parent(s): 2e0a0ba

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
@@ -1228,9 +1228,9 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1228
  print('shape 3.8: ', clip_image_embedse.shape)
1229
  # 4. Concatenate the zeros and the average embedding
1230
  clip_image_embeds2 = torch.cat([zeros_tensor, average_image_embed], dim=0)
1231
- print('shape 4: ', clip_image_embeds.shape)
1232
  clip_image_embeds = torch.cat([zeros_tensora, clip_image_embedsa], dim=0)
1233
- print('shape 4a: ', clip_image_embeds2.shape)
1234
 
1235
  '''
1236
  #clip_image_embeds = clip_image_embeds.unsqueeze(0) # Add a dimension at the beginning so now you have [1, 2*seq_len_img, embed_dim_img]
 
1228
  print('shape 3.8: ', clip_image_embedse.shape)
1229
  # 4. Concatenate the zeros and the average embedding
1230
  clip_image_embeds2 = torch.cat([zeros_tensor, average_image_embed], dim=0)
1231
+ print('shape 4: ', clip_image_embeds2.shape)
1232
  clip_image_embeds = torch.cat([zeros_tensora, clip_image_embedsa], dim=0)
1233
+ print('shape 4a: ', clip_image_embeds.shape)
1234
 
1235
  '''
1236
  #clip_image_embeds = clip_image_embeds.unsqueeze(0) # Add a dimension at the beginning so now you have [1, 2*seq_len_img, embed_dim_img]