Spaces:
Runtime error
Runtime error
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 |
|