jytole commited on
Commit
70f1ffe
·
1 Parent(s): b1e754f

Change to DDPM Scheduler

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -1,9 +1,12 @@
1
  import gradio as gr
2
 
3
  from diffusers import AudioLDMPipeline
 
4
  #from diffusers import DPMSolverMultistepScheduler
 
5
  #from diffusers import DDIMScheduler
6
- from diffusers import DDIMInverseScheduler
 
7
 
8
  from transformers import AutoProcessor, ClapModel
9
 
@@ -16,7 +19,8 @@ repo_id = "cvssp/audioldm-s-full-v2"
16
  pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
17
  #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
18
  #pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
19
- pipe.scheduler = DDIMInverseScheduler.from_config(pipe.scheduler.config)
 
20
  pipe = pipe.to(device)
21
 
22
  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
+ # The recommended "fast" scheduler:
5
  #from diffusers import DPMSolverMultistepScheduler
6
+ # The Default AudioLDM scheduler:
7
  #from diffusers import DDIMScheduler
8
+ #from diffusers import DDIMInverseScheduler
9
+ from diffusers import DDPMScheduler
10
 
11
  from transformers import AutoProcessor, ClapModel
12
 
 
19
  pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
20
  #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
21
  #pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
22
+ #pipe.scheduler = DDIMInverseScheduler.from_config(pipe.scheduler.config)
23
+ pipe.scheduler = DDPMScheduler.from_config(pipe.scheduler.config)
24
  pipe = pipe.to(device)
25
 
26
  clap_model = ClapModel.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-full").to(device)