jytole commited on
Commit
eb1b51b
1 Parent(s): a76a8ba

Update app.py to change scheduler

Browse files

Changed scheduler from default (DDIM) to DPM Multistep

Files changed (1) hide show
  1. app.py +1 -0
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):