Change to Heun Scheduler

app.py

@@ -2,13 +2,13 @@ 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 HeunDiscreteScheduler
+from diffusers import HeunDiscreteScheduler
 #from diffusers import KDPM2DiscreteScheduler
 #from diffusers import KDPM2AncestralDiscreteScheduler
 #from diffusers import LMSDiscreteScheduler
@@ -26,13 +26,13 @@ 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 = HeunDiscreteScheduler.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)
 #pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)