1inkusFace commited on
Commit
f9c67d1
·
verified ·
1 Parent(s): 95b1b23

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +3 -1
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1217,10 +1217,12 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1217
  print('shape 2a: ', average_image_embedf.shape)
1218
  # 3. Create a tensor of zeros with the same shape as the averaged embedding
1219
  zeros_tensor = torch.zeros_like(average_image_embed)
 
1220
  print('shape 3: ', zeros_tensor.shape)
 
1221
  # 4. Concatenate the zeros and the average embedding
1222
  clip_image_embeds = torch.cat([zeros_tensor, average_image_embed], dim=0)
1223
- clip_image_embeds2 = torch.cat([zeros_tensor, average_image_embedf], dim=0)
1224
  print('shape 4: ', clip_image_embeds.shape)
1225
  print('shape 4a: ', clip_image_embeds2.shape)
1226
  '''
 
1217
  print('shape 2a: ', average_image_embedf.shape)
1218
  # 3. Create a tensor of zeros with the same shape as the averaged embedding
1219
  zeros_tensor = torch.zeros_like(average_image_embed)
1220
+ zeros_tensora = torch.zeros_like(average_image_embed)
1221
  print('shape 3: ', zeros_tensor.shape)
1222
+ print('shape 3a: ', zeros_tensora.shape)
1223
  # 4. Concatenate the zeros and the average embedding
1224
  clip_image_embeds = torch.cat([zeros_tensor, average_image_embed], dim=0)
1225
+ clip_image_embeds2 = torch.cat([zeros_tensora, average_image_embedf], dim=0)
1226
  print('shape 4: ', clip_image_embeds.shape)
1227
  print('shape 4a: ', clip_image_embeds2.shape)
1228
  '''