Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
6af5476
1
Parent(s):
00c43c3
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 |
)
|