Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -580,7 +580,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
580 |
|
581 |
if self.do_classifier_free_guidance:
|
582 |
prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds], dim=0)
|
583 |
-
pooled_prompt_embeds = torch.cat([
|
584 |
|
585 |
# 4. Prepare latent variables
|
586 |
num_channels_latents = self.transformer.config.in_channels // 4
|
|
|
580 |
|
581 |
if self.do_classifier_free_guidance:
|
582 |
prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds], dim=0)
|
583 |
+
pooled_prompt_embeds = torch.cat([negative_prompt_embeds, pooled_prompt_embeds], dim=0)
|
584 |
|
585 |
# 4. Prepare latent variables
|
586 |
num_channels_latents = self.transformer.config.in_channels // 4
|