Update app.py

app.py

@@ -140,7 +140,7 @@ def train_step(state, batch, rng):
             {'params': params["unet"]},
             noisy_latents,
             timesteps,
-            encoder_hidden_states=encoder_hidden_states,
+            encoder_hidden_states,
             train=True,
         )