Hugging Face's logo

Spaces:
fluorescenrgrey
/
midi-audioldm
Runtime error

App Files Community
lauraibnz commited on
Commit
a475156
•
1 Parent(s): c7db663

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ else:
16
  torch_dtype = torch.float32
17
 
18
  controlnet = ControlNetModel.from_pretrained(
19
- "lauraibnz/midi-audioldm", torch_dtype=torch_dtype)
20
  pipe = AudioLDMControlNetPipeline.from_pretrained(
21
  "cvssp/audioldm-m-full", controlnet=controlnet, torch_dtype=torch_dtype)
22
  pipe = pipe.to(device)
 
16
  torch_dtype = torch.float32
17
 
18
  controlnet = ControlNetModel.from_pretrained(
19
+ "lauraibnz/midi-audioldm-v2", torch_dtype=torch_dtype)
20
  pipe = AudioLDMControlNetPipeline.from_pretrained(
21
  "cvssp/audioldm-m-full", controlnet=controlnet, torch_dtype=torch_dtype)
22
  pipe = pipe.to(device)