1inkusFace commited on
Commit
2520157
·
verified ·
1 Parent(s): a943a51

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +1 -1
pipeline_stable_diffusion_3_ipa.py CHANGED
@@ -1178,7 +1178,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
1178
  ## clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0)
1179
  clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0) #.unsqueeze(0)
1180
  bs_embed, seq_len = clip_image_embeds.shape
1181
- clip_image_embeds = clip_image_embeds.view(bs_embed, seq_len, -1) # Simplified reshape
1182
 
1183
  # 4. Prepare timesteps
1184
  timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, timesteps)
 
1178
  ## clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0)
1179
  clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0) #.unsqueeze(0)
1180
  bs_embed, seq_len = clip_image_embeds.shape
1181
+ clip_image_embeds = clip_image_embeds.view(bs_embed, seq_len) # Simplified reshape
1182
 
1183
  # 4. Prepare timesteps
1184
  timesteps, num_inference_steps = retrieve_timesteps(self.scheduler, num_inference_steps, device, timesteps)