Update demos/musicgen_app.py
Browse files- demos/musicgen_app.py +1 -1
demos/musicgen_app.py
CHANGED
|
@@ -142,7 +142,7 @@ def load_model(version="facebook/musicgen-small"):
|
|
| 142 |
torch.cuda.empty_cache()
|
| 143 |
MODEL = None # in case loading would crash
|
| 144 |
# MODEL = MusicGen.get_pretrained("/Users/ebenge/repos/audiocraft/data/")
|
| 145 |
-
lm = load_lm_model("
|
| 146 |
compression_model = load_compression_model(
|
| 147 |
"facebook/encodec_32khz", device="cudu"
|
| 148 |
)
|
|
|
|
| 142 |
torch.cuda.empty_cache()
|
| 143 |
MODEL = None # in case loading would crash
|
| 144 |
# MODEL = MusicGen.get_pretrained("/Users/ebenge/repos/audiocraft/data/")
|
| 145 |
+
lm = load_lm_model("data/state_dict.bin", device="cudu")
|
| 146 |
compression_model = load_compression_model(
|
| 147 |
"facebook/encodec_32khz", device="cudu"
|
| 148 |
)
|