Update app.py

app.py

@@ -1,4 +1,4 @@
-from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
+from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler, DPMSolverMultistepScheduler
 import torch
 from transformers import pipeline
 import gradio as gr
@@ -20,7 +20,7 @@ InferenceClient=InferenceClient()
 apol=[]
 
 pipe = accelerator.prepare(StableDiffusionXLPipeline.from_pretrained("Lykon/dreamshaper-xl-lightning", torch_dtype=torch.bfloat16, use_safetensors=True, variant=None, safety_checker=False))
-pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.unet.to(memory_format=torch.channels_last)
 pipe = accelerator.prepare(pipe.to("cpu"))