Change to DPM Single
Browse files
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 |
-
|
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 |
-
|
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)
|