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

Change to DPM Single

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -6,8 +6,8 @@ from diffusers import AudioLDMPipeline
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
13
  #from diffusers import KDPM2AncestralDiscreteScheduler
@@ -29,8 +29,8 @@ 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)
36
  #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
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
13
  #from diffusers import KDPM2AncestralDiscreteScheduler
 
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)
36
  #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)