jytole commited on
Commit
e33859f
·
1 Parent(s): 90e39db

Change to PNDM Scheduler

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,8 +11,8 @@ from diffusers import AudioLDMPipeline
11
  #from diffusers import HeunDiscreteScheduler
12
  #from diffusers import KDPM2DiscreteScheduler
13
  #from diffusers import KDPM2AncestralDiscreteScheduler
14
- from diffusers import LMSDiscreteScheduler
15
- #from diffusers import PNDMScheduler
16
  #from diffusers import EulerDiscreteScheduler
17
  #from diffusers import EulerAncestralDiscreteScheduler
18
  #from diffusers import UniPCMultistepScheduler
@@ -35,8 +35,8 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
35
  #pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config)
36
  #pipe.scheduler = KDPM2DiscreteScheduler.from_config(pipe.scheduler.config)
37
  #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
38
- pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
39
- #pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
40
  #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
41
  #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
42
  #pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
11
  #from diffusers import HeunDiscreteScheduler
12
  #from diffusers import KDPM2DiscreteScheduler
13
  #from diffusers import KDPM2AncestralDiscreteScheduler
14
+ #from diffusers import LMSDiscreteScheduler
15
+ from diffusers import PNDMScheduler
16
  #from diffusers import EulerDiscreteScheduler
17
  #from diffusers import EulerAncestralDiscreteScheduler
18
  #from diffusers import UniPCMultistepScheduler
 
35
  #pipe.scheduler = HeunDiscreteScheduler.from_config(pipe.scheduler.config)
36
  #pipe.scheduler = KDPM2DiscreteScheduler.from_config(pipe.scheduler.config)
37
  #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
38
+ #pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
39
+ pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
40
  #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
41
  #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
42
  #pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)