Update app.py

app.py

@@ -111,7 +111,7 @@ def scheduler_swap_callback(pipeline, step_index, timestep, callback_kwargs):
         #pipe.scheduler.set_timesteps(num_inference_steps*.70)
       #  print(f"-- setting step {pipeline.num_timesteps * 0.9} --")
       #  pipeline.scheduler._step_index = pipeline.num_timesteps * 0.9
-    return
+    return {"latents": callback_kwargs["latents"]}   
 
 
 def load_and_prepare_model():