Update app.py

app.py

@@ -120,7 +120,7 @@ def load_and_prepare_model(model_id):
         use_safetensors=True,
         vae=vae,
     )
-    sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="squaredcos_cap_v2",use_karras_sigmas=True, algorithm_type="dpmsolver++")
+    sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear",use_karras_sigmas=True, algorithm_type="dpmsolver++")
     #sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="linear", algorithm_type="dpmsolver++")
     #sched = DDIMScheduler.from_config(pipe.scheduler.config)
     pipe.scheduler=sched
@@ -191,6 +191,7 @@ def generate_60(
     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()