jytole commited on
Commit
bd535b8
·
1 Parent(s): 5df73fe

Explicitly set DDIMScheduler

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
 
3
- from diffusers import AudioLDMPipeline, DPMSolverMultistepScheduler
 
 
4
 
5
  from transformers import AutoProcessor, ClapModel
6
 
@@ -12,6 +14,7 @@ device="cpu"
12
  repo_id = "cvssp/audioldm-s-full-v2"
13
  pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
14
  #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
15
  pipe = pipe.to(device)
16
 
17
  clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)
 
1
  import gradio as gr
2
 
3
+ from diffusers import AudioLDMPipeline
4
+ #from diffusers import DPMSolverMultistepScheduler
5
+ from diffusers import DDIMScheduler
6
 
7
  from transformers import AutoProcessor, ClapModel
8
 
 
14
  repo_id = "cvssp/audioldm-s-full-v2"
15
  pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
16
  #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
17
+ pipe.scheduler = DDIMSolverMultistepScheduler.from_config(pipe.scheduler.config)
18
  pipe = pipe.to(device)
19
 
20
  clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)