feat: change model used

2024-07-20 12:42:35 +01:00
parent 1cc81f4da2
commit f1ecddc918
1 changed files with 1 additions and 1 deletions
--- a/generate.py
+++ b/generate.py
@@ -2,7 +2,7 @@ import torchaudio
 from audiocraft.models.magnet import MAGNeT
 from audiocraft.data.audio import audio_write

-MODEL_NAME = "facebook/magnet-small-10secs"
+MODEL_NAME = "facebook/magnet-medium-30secs"

 print(f"getting {MODEL_NAME}...")