Update app.py

app.py

@@ -242,6 +242,7 @@ def generate_90(
     if use_resolution_binning:
         options["use_resolution_binning"] = True
     images = []
+    pipe.scheduler.set_timesteps(num_inference_steps,device)
     with torch.no_grad():
       for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()