update schulder

app.py

@@ -5,7 +5,10 @@ import torch.distributed as dist
 import torch.multiprocessing as mp
 import argparse
 
-from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
+#from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
+
+
 from tqdm.auto import tqdm
 import random
 import gradio as gr
@@ -129,7 +132,8 @@ if __name__ == "__main__":
         dtype = torch.float32  # CPU does not support float16, use float32 instead
 
 
+    dpm = DPMSolverMultistepScheduler.from_pretrained(PATH, subfolder="scheduler")
+    pipe = StableDiffusionPipeline.from_single_file(PATH,torch_dtype=dtype, scheduler=dpm).to(device)
     
-    pipe = StableDiffusionPipeline.from_single_file(PATH,torch_dtype=dtype).to(device)
     demo.queue(concurrency_count=2,
             ).launch()