Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
2580ebe
1
Parent(s):
ab87e03
Update pipeline.py
Browse files- 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 |
|