Change to DDIM Scheduler

app.py

@@ -4,24 +4,18 @@ from diffusers import AudioLDMPipeline
 # The recommended "fast" scheduler:
 #from diffusers import DPMSolverMultistepScheduler
 # The Default AudioLDM scheduler:
-#from diffusers import DDIMScheduler
-#from diffusers import DDIMInverseScheduler
+from diffusers import DDIMScheduler
 #from diffusers import DDPMScheduler
 #from diffusers import DEISMultistepScheduler
 #from diffusers import DPMSolverSinglestepScheduler
 #from diffusers import HeunDiscreteScheduler
 #from diffusers import KDPM2DiscreteScheduler
 #from diffusers import KDPM2AncestralDiscreteScheduler
-#from diffusers import KarrasVeScheduler
 #from diffusers import LMSDiscreteScheduler
 #from diffusers import PNDMScheduler
-#from diffusers import ScoreSdeVeScheduler
-#from diffusers import IPNDMScheduler
 #from diffusers import EulerDiscreteScheduler
 #from diffusers import EulerAncestralDiscreteScheduler
-#from diffusers import VQDiffusionScheduler
 #from diffusers import UniPCMultistepScheduler
-from diffusers import RePaintScheduler
 
 from transformers import AutoProcessor, ClapModel
 
@@ -33,24 +27,18 @@ 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 = DDIMScheduler.from_config(pipe.scheduler.config)
-#pipe.scheduler = DDIMInverseScheduler.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 = KDPM2DiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-#pipe.scheduler = KarrasVeScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
-#pipe.scheduler = ScoreSdeVeScheduler.from_config(pipe.scheduler.config)
-#pipe.scheduler = IPNDMScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-#pipe.scheduler = VQDiffusionScheduler.from_config(pipe.scheduler.config)
 #pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.scheduler = RePaintScheduler.from_config(pipe.scheduler.config)
 pipe = pipe.to(device)
 
 clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)