InstantX
/

flux-dev-de-distill-diffusers

Model card Files Files and versions

wanghaofan commited on Oct 15, 2024

Commit

9029417

·

verified ·

1 Parent(s): b8c1258

Update pipeline_flux_de_distill.py

Files changed (1) hide show

pipeline_flux_de_distill.py +1 -1

pipeline_flux_de_distill.py CHANGED Viewed

@@ -739,7 +739,7 @@ class FluxPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFileMixin):
                 timestep = t.expand(latent_model_input.shape[0])
                 noise_pred = self.transformer(
-                    hidden_states=latents,
                     timestep=timestep / 1000,
                     pooled_projections=pooled_prompt_embeds,
                     encoder_hidden_states=prompt_embeds,

                 timestep = t.expand(latent_model_input.shape[0])
                 noise_pred = self.transformer(
+                    hidden_states=latent_model_input,
                     timestep=timestep / 1000,
                     pooled_projections=pooled_prompt_embeds,
                     encoder_hidden_states=prompt_embeds,