jytole commited on
Commit
d7cdc95
·
1 Parent(s): 15f4e52

Change to DDIM Scheduler

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,8 +4,8 @@ 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 DDPMScheduler
9
  #from diffusers import DEISMultistepScheduler
10
  #from diffusers import DPMSolverSinglestepScheduler
11
  #from diffusers import HeunDiscreteScheduler
@@ -27,8 +27,8 @@ device="cpu"
27
  repo_id = "cvssp/audioldm-s-full-v2"
28
  pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
29
  #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
30
- #pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
31
- pipe.scheduler = DDPMScheduler.from_config(pipe.scheduler.config)
32
  #pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
33
  #pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config)
34
  #pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config)
 
4
  # The recommended "fast" scheduler:
5
  #from diffusers import DPMSolverMultistepScheduler
6
  # The Default AudioLDM scheduler:
7
+ from diffusers import DDIMScheduler
8
+ #from diffusers import DDPMScheduler
9
  #from diffusers import DEISMultistepScheduler
10
  #from diffusers import DPMSolverSinglestepScheduler
11
  #from diffusers import HeunDiscreteScheduler
 
27
  repo_id = "cvssp/audioldm-s-full-v2"
28
  pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
29
  #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
30
+ pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
31
+ #pipe.scheduler = DDPMScheduler.from_config(pipe.scheduler.config)
32
  #pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
33
  #pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config)
34
  #pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config)