Spaces:
Runtime error
Runtime error
Update pipeline.py
Browse files- pipeline.py +3 -2
pipeline.py
CHANGED
|
@@ -277,7 +277,6 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
| 277 |
prompt=negative_prompt_2,
|
| 278 |
device=device,
|
| 279 |
num_images_per_prompt=num_images_per_prompt,
|
| 280 |
-
max_sequence_length=max_sequence_length,
|
| 281 |
)
|
| 282 |
|
| 283 |
negative_clip_prompt_embed = torch.nn.functional.pad(
|
|
@@ -285,7 +284,9 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
| 285 |
(0, t5_negative_prompt_embed.shape[-1] - negative_clip_prompt_embed.shape[-1]),
|
| 286 |
)
|
| 287 |
|
| 288 |
-
negative_prompt_embeds = torch.cat([negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-
|
|
|
|
|
|
|
| 289 |
negative_pooled_prompt_embeds = torch.cat(
|
| 290 |
[negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-1
|
| 291 |
)
|
|
|
|
| 277 |
prompt=negative_prompt_2,
|
| 278 |
device=device,
|
| 279 |
num_images_per_prompt=num_images_per_prompt,
|
|
|
|
| 280 |
)
|
| 281 |
|
| 282 |
negative_clip_prompt_embed = torch.nn.functional.pad(
|
|
|
|
| 284 |
(0, t5_negative_prompt_embed.shape[-1] - negative_clip_prompt_embed.shape[-1]),
|
| 285 |
)
|
| 286 |
|
| 287 |
+
negative_prompt_embeds = torch.cat([negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-1)
|
| 288 |
+
|
| 289 |
+
|
| 290 |
negative_pooled_prompt_embeds = torch.cat(
|
| 291 |
[negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-1
|
| 292 |
)
|