Spaces:

descript
/

vampnet

Sleeping

Hugo Flores Garcia commited on Jul 11, 2023

Commit

a2a8108

1 Parent(s): 8b2a6b0

wavebeat

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,6 +22,7 @@ interface = Interface(
     coarse_ckpt="./models/vampnet/coarse.pth",
     coarse2fine_ckpt="./models/vampnet/c2f.pth",
     codec_ckpt="./models/vampnet/codec.pth",
     device="cuda" if torch.cuda.is_available() else "cpu",
 )

     coarse_ckpt="./models/vampnet/coarse.pth",
     coarse2fine_ckpt="./models/vampnet/c2f.pth",
     codec_ckpt="./models/vampnet/codec.pth",
+    wavebeat_ckpt="./models/wavebeat.pth",
     device="cuda" if torch.cuda.is_available() else "cpu",
 )