Update app.py to change scheduler
Browse filesChanged scheduler from default (DDIM) to DPM Multistep
app.py
CHANGED
@@ -6,6 +6,7 @@ import torch
|
|
6 |
|
7 |
repo_id = "cvssp/audioldm-s-full-v2"
|
8 |
pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
|
|
|
9 |
pipe = pipe.to("cpu")
|
10 |
|
11 |
def texttoaudio(prompt):
|
|
|
6 |
|
7 |
repo_id = "cvssp/audioldm-s-full-v2"
|
8 |
pipe = AudioLDMPipeline.from_pretrained(repo_id, torch_dtype=torch.float32)
|
9 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
10 |
pipe = pipe.to("cpu")
|
11 |
|
12 |
def texttoaudio(prompt):
|