LibreFLUX_LoRAs_Gallery

Runtime error

AlekseyCalvin commited on Oct 19, 2024

Commit

57f9be9

verified ·

1 Parent(s): 575de74

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -628,7 +628,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
                     return_dict=False,
                 )[0]
                 noise_pred_uncond = self.transformer(
-                    hidden_states=latents,
                     timestep=timestep / 1000,
                     guidance=guidance,
                     pooled_projections=negative_pooled_prompt_embeds,

                     return_dict=False,
                 )[0]
                 noise_pred_uncond = self.transformer(
+                    hidden_states=latent_model_input,
                     timestep=timestep / 1000,
                     guidance=guidance,
                     pooled_projections=negative_pooled_prompt_embeds,