Change to DPM Multistep

app.py

@@ -2,12 +2,12 @@ import gradio as gr
 
 from diffusers import AudioLDMPipeline
 # The recommended "fast" scheduler:
-#from diffusers import DPMSolverMultistepScheduler
+from diffusers import DPMSolverMultistepScheduler
 # The Default AudioLDM scheduler:
 #from diffusers import DDIMScheduler
 #from diffusers import DDPMScheduler
 #from diffusers import DEISMultistepScheduler
-from diffusers import DPMSolverSinglestepScheduler
+#from diffusers import DPMSolverSinglestepScheduler
 #from diffusers import HeunDiscreteScheduler
 #from diffusers import KDPM2DiscreteScheduler
 #from diffusers import KDPM2AncestralDiscreteScheduler
@@ -26,11 +26,12 @@ device="cpu"
 
 repo_id = "cvssp/audioldm-s-full-v2"
 pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
-#pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = DDPMScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config)
+#pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config)
+
 #pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = KDPM2DiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)