AlekseyCalvin commited on
Commit
2580ebe
1 Parent(s): ab87e03

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +7 -6
pipeline.py CHANGED
@@ -558,22 +558,23 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
558
  )
559
  (
560
  prompt_embeds,
561
- negative_prompt_embeds,
562
  pooled_prompt_embeds,
 
 
563
  negative_pooled_prompt_embeds,
564
  ) = self.encode_prompt(
565
  prompt=prompt,
566
  prompt_2=prompt_2,
 
 
 
 
567
  negative_prompt=negative_prompt,
568
  negative_prompt_2=negative_prompt_2,
569
- do_classifier_free_guidance=self.do_classifier_free_guidance,
570
  prompt_embeds=prompt_embeds,
571
- pooled_prompt_embeds=pooled_prompt_embeds,
572
  negative_prompt_embeds=negative_prompt_embeds,
 
573
  negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
574
- device=device,
575
- num_images_per_prompt=num_images_per_prompt,
576
- max_sequence_length=max_sequence_length,
577
  lora_scale=lora_scale,
578
  )
579
 
 
558
  )
559
  (
560
  prompt_embeds,
 
561
  pooled_prompt_embeds,
562
+ text_ids,
563
+ negative_prompt_embeds,
564
  negative_pooled_prompt_embeds,
565
  ) = self.encode_prompt(
566
  prompt=prompt,
567
  prompt_2=prompt_2,
568
+ num_images_per_prompt=num_images_per_prompt,
569
+ max_sequence_length=max_sequence_length,
570
+ do_classifier_free_guidance=self.do_classifier_free_guidance,
571
+ device=device,
572
  negative_prompt=negative_prompt,
573
  negative_prompt_2=negative_prompt_2,
 
574
  prompt_embeds=prompt_embeds,
 
575
  negative_prompt_embeds=negative_prompt_embeds,
576
+ pooled_prompt_embeds=pooled_prompt_embeds,
577
  negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
 
 
 
578
  lora_scale=lora_scale,
579
  )
580