Update demos/musicgen_app.py
Browse files- demos/musicgen_app.py +1 -1
demos/musicgen_app.py
CHANGED
|
@@ -101,7 +101,7 @@ def load_model(version="facebook/musicgen-small"):
|
|
| 101 |
del MODEL
|
| 102 |
torch.cuda.empty_cache()
|
| 103 |
MODEL = None # in case loading would crash
|
| 104 |
-
MODEL = MusicGen.get_pretrained("
|
| 105 |
print("Custom model loaded.")
|
| 106 |
|
| 107 |
|
|
|
|
| 101 |
del MODEL
|
| 102 |
torch.cuda.empty_cache()
|
| 103 |
MODEL = None # in case loading would crash
|
| 104 |
+
MODEL = MusicGen.get_pretrained("../data")
|
| 105 |
print("Custom model loaded.")
|
| 106 |
|
| 107 |
|