Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -286,7 +286,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
286 |
(0, t5_negative_prompt_embeds.shape[-1] - negative_pooled_prompt_embeds.shape[-1]),
|
287 |
)
|
288 |
|
289 |
-
negative_prompt_embeds = torch.cat([
|
290 |
|
291 |
|
292 |
if self.text_encoder is not None:
|
|
|
286 |
(0, t5_negative_prompt_embeds.shape[-1] - negative_pooled_prompt_embeds.shape[-1]),
|
287 |
)
|
288 |
|
289 |
+
negative_prompt_embeds = torch.cat([negative_pooled_prompt_embeds, t5_negative_prompt_embeds], dim=-2)
|
290 |
|
291 |
|
292 |
if self.text_encoder is not None:
|