jytole commited on
Commit
a8d1ef8
·
1 Parent(s): 65614df

Change ScoreSdeVeScheduler

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -14,8 +14,8 @@ from diffusers import AudioLDMPipeline
14
  #from diffusers import KDPM2AncestralDiscreteScheduler
15
  #from diffusers import KarrasVeScheduler
16
  #from diffusers import LMSDiscreteScheduler
17
- from diffusers import PNDMScheduler
18
- #from diffusers import ScoreSdeVeScheduler
19
  #from diffusers import IPNDMScheduler
20
  #from diffusers import ScoreSdeVpScheduler
21
  #from diffusers import EulerDiscreteScheduler
@@ -44,8 +44,8 @@ pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
44
  #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
45
  #pipe.scheduler = KarrasVeScheduler.from_config(pipe.scheduler.config)
46
  #pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
47
- pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
48
- #pipe.scheduler = ScoreSdeVeScheduler.from_config(pipe.scheduler.config)
49
  #pipe.scheduler = IPNDMScheduler.from_config(pipe.scheduler.config)
50
  #pipe.scheduler = ScoreSdeVpScheduler.from_config(pipe.scheduler.config)
51
  #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 
14
  #from diffusers import KDPM2AncestralDiscreteScheduler
15
  #from diffusers import KarrasVeScheduler
16
  #from diffusers import LMSDiscreteScheduler
17
+ #from diffusers import PNDMScheduler
18
+ from diffusers import ScoreSdeVeScheduler
19
  #from diffusers import IPNDMScheduler
20
  #from diffusers import ScoreSdeVpScheduler
21
  #from diffusers import EulerDiscreteScheduler
 
44
  #pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
45
  #pipe.scheduler = KarrasVeScheduler.from_config(pipe.scheduler.config)
46
  #pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
47
+ #pipe.scheduler = PNDMScheduler.from_config(pipe.scheduler.config)
48
+ pipe.scheduler = ScoreSdeVeScheduler.from_config(pipe.scheduler.config)
49
  #pipe.scheduler = IPNDMScheduler.from_config(pipe.scheduler.config)
50
  #pipe.scheduler = ScoreSdeVpScheduler.from_config(pipe.scheduler.config)
51
  #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)