jytole commited on
Commit
67b8718
·
1 Parent(s): d7cdc95

Change to DEIS Scheduler

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,9 +4,9 @@ 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
12
  #from diffusers import KDPM2DiscreteScheduler
@@ -27,9 +27,9 @@ 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)
35
  #pipe.scheduler = KDPM2DiscreteScheduler.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
12
  #from diffusers import KDPM2DiscreteScheduler
 
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)
35
  #pipe.scheduler = KDPM2DiscreteScheduler.from_config(pipe.scheduler.config)