Change ScoreSdeVeScheduler
Browse files
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 |
-
|
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 |
-
|
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)
|