Update pipeline_flux_de_distill.py

pipeline_flux_de_distill.py

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