1inkusFace commited on
Commit
1023f95
·
verified ·
1 Parent(s): 08bea07

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
@@ -1193,10 +1193,10 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1193
  '''
1194
  bs_embed, seq_len, _ = concatenated_embeds.shape
1195
  # batch_size = concatenated_embeds.shape[0]
1196
- image_prompt_embeds = image_prompt_embeds.repeat(1, 1, 1)
1197
- reshaped_embeds = concatenated_embeds.view(bs_embed, seq_len, -1)
1198
  # Pass the concatenated embeddings through the linear layer
1199
- clip_image_embeds = linear_layer(reshaped_embeds)
1200
 
1201
  # Add a ReLU activation for non-linearity (optional)
1202
  #combined_embeds = torch.relu(combined_embeds)
 
1193
  '''
1194
  bs_embed, seq_len, _ = concatenated_embeds.shape
1195
  # batch_size = concatenated_embeds.shape[0]
1196
+ concatenated_embeds = concatenated_embeds.repeat(1, 1, 1)
1197
+ concatenated_embeds = concatenated_embeds.view(bs_embed, seq_len, -1)
1198
  # Pass the concatenated embeddings through the linear layer
1199
+ clip_image_embeds = linear_layer(concatenated_embeds)
1200
 
1201
  # Add a ReLU activation for non-linearity (optional)
1202
  #combined_embeds = torch.relu(combined_embeds)