Change to KDPM2Discrete

app.py

@@ -9,13 +9,13 @@ from diffusers import AudioLDMPipeline
 #from diffusers import DEISMultistepScheduler
 #from diffusers import DPMSolverSinglestepScheduler
 #from diffusers import HeunDiscreteScheduler
-#from diffusers import KDPM2DiscreteScheduler
+from diffusers import KDPM2DiscreteScheduler
 #from diffusers import KDPM2AncestralDiscreteScheduler
 #from diffusers import LMSDiscreteScheduler
 #from diffusers import PNDMScheduler
 #from diffusers import EulerDiscreteScheduler
 #from diffusers import EulerAncestralDiscreteScheduler
-from diffusers import UniPCMultistepScheduler
+#from diffusers import UniPCMultistepScheduler
 
 from transformers import AutoProcessor, ClapModel
 
@@ -33,13 +33,13 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
 #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 = KDPM2DiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
+#pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 pipe = pipe.to(device)
 
 clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)