Update app.py

app.py

@@ -10,7 +10,7 @@ import numpy as np
 import PIL.Image
 import spaces
 import torch
-from diffusers import AutoencoderKL, DiffusionPipeline, EulerAncestralDiscreteScheduler
+from diffusers import AutoencoderKL, DiffusionPipeline, DPMSolverMultistepScheduler
 
 DESCRIPTION = "# SDXL"
 if not torch.cuda.is_available():
@@ -61,11 +61,11 @@ def generate(
     if torch.cuda.is_available():
 
         if not use_vae:  
-            scheduler = EulerAncestralDiscreteScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
+            scheduler =  DPMSolverMultistepScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
             pipe = DiffusionPipeline.from_pretrained(dropdown_model, scheduler=scheduler, torch_dtype=torch.float16)
             
         if use_vae:
-            scheduler = EulerAncestralDiscreteScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
+            scheduler =  DPMSolverMultistepScheduler.from_pretrained(dropdown_model, subfolder="scheduler")
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16, variant="fp16")
             pipe = DiffusionPipeline.from_pretrained(dropdown_model, vae=vae, scheduler=scheduler, torch_dtype=torch.float16)