vampnet

Sleeping

Hugo Flores Garcia commited on Jul 11, 2023

Commit

b1581a7

•

1 Parent(s): 49febdc

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",
 )