Change to DEIS Scheduler
Browse files
app.py
CHANGED
|
@@ -4,9 +4,9 @@ from diffusers import AudioLDMPipeline
|
|
| 4 |
# The recommended "fast" scheduler:
|
| 5 |
#from diffusers import DPMSolverMultistepScheduler
|
| 6 |
# The Default AudioLDM scheduler:
|
| 7 |
-
from diffusers import DDIMScheduler
|
| 8 |
#from diffusers import DDPMScheduler
|
| 9 |
-
|
| 10 |
#from diffusers import DPMSolverSinglestepScheduler
|
| 11 |
#from diffusers import HeunDiscreteScheduler
|
| 12 |
#from diffusers import KDPM2DiscreteScheduler
|
|
@@ -27,9 +27,9 @@ device="cpu"
|
|
| 27 |
repo_id = "cvssp/audioldm-s-full-v2"
|
| 28 |
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 |
-
|
| 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)
|
|
|
|
| 4 |
# The recommended "fast" scheduler:
|
| 5 |
#from diffusers import DPMSolverMultistepScheduler
|
| 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
|
|
|
|
| 27 |
repo_id = "cvssp/audioldm-s-full-v2"
|
| 28 |
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)
|