Spaces:
Runtime error
Runtime error
crypto-code
commited on
Commit
•
968cca5
1
Parent(s):
0038c38
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ import av
|
|
17 |
import subprocess
|
18 |
import librosa
|
19 |
|
20 |
-
args = {"model": "./ckpts/
|
21 |
"mert_path": "m-a-p/MERT-v1-330M", "vit_path": "google/vit-base-patch16-224", "vivit_path": "google/vivit-b-16x2-kinetics400",
|
22 |
"music_decoder": "musicgen", "music_decoder_path": "facebook/musicgen-medium"}
|
23 |
|
|
|
17 |
import subprocess
|
18 |
import librosa
|
19 |
|
20 |
+
args = {"model": "./ckpts/checkpoint.pth", "llama_type": "7B", "llama_dir": "./ckpts/LLaMA-2",
|
21 |
"mert_path": "m-a-p/MERT-v1-330M", "vit_path": "google/vit-base-patch16-224", "vivit_path": "google/vivit-b-16x2-kinetics400",
|
22 |
"music_decoder": "musicgen", "music_decoder_path": "facebook/musicgen-medium"}
|
23 |
|