AlekseyCalvin commited on
Commit
db6571d
·
verified ·
1 Parent(s): 2580ebe

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -580,7 +580,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
580
 
581
  if self.do_classifier_free_guidance:
582
  prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds], dim=0)
583
- pooled_prompt_embeds = torch.cat([negative_pooled_prompt_embeds, pooled_prompt_embeds], dim=0)
584
 
585
  # 4. Prepare latent variables
586
  num_channels_latents = self.transformer.config.in_channels // 4
 
580
 
581
  if self.do_classifier_free_guidance:
582
  prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds], dim=0)
583
+ pooled_prompt_embeds = torch.cat([negative_prompt_embeds, pooled_prompt_embeds], dim=0)
584
 
585
  # 4. Prepare latent variables
586
  num_channels_latents = self.transformer.config.in_channels // 4