Change to DDIMInverse

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 
 from diffusers import AudioLDMPipeline
 #from diffusers import DPMSolverMultistepScheduler
-from diffusers import DDIMScheduler
+#from diffusers import DDIMScheduler
+from diffusers import DDIMInverseScheduler
 
 from transformers import AutoProcessor, ClapModel
 
@@ -14,7 +15,8 @@ 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 = DDIMScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = DDIMInverseScheduler.from_config(pipe.scheduler.config)
 pipe = pipe.to(device)
 
 clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)