From f1ecddc918078a3f678a6d3820d38185123d881c Mon Sep 17 00:00:00 2001 From: Kenneth Date: Sat, 20 Jul 2024 12:42:35 +0100 Subject: [PATCH] feat: change model used --- generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generate.py b/generate.py index 74818c1..de24ec7 100644 --- 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}...")