From ec71c0b235920d594bd414f3e659f57e6bb4a885 Mon Sep 17 00:00:00 2001 From: Kenneth Date: Sat, 20 Jul 2024 14:46:25 +0100 Subject: [PATCH] feat: change model to musicgen --- generate.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/generate.py b/generate.py index de24ec7..cce44ed 100644 --- a/generate.py +++ b/generate.py @@ -1,12 +1,12 @@ import torchaudio -from audiocraft.models.magnet import MAGNeT +from audiocraft.models.musicgen import MusicGen from audiocraft.data.audio import audio_write -MODEL_NAME = "facebook/magnet-medium-30secs" +MODEL_NAME = "facebook/musicgen-large" print(f"getting {MODEL_NAME}...") -model = MAGNeT.get_pretrained(MODEL_NAME) +model = MusicGen.get_pretrained(MODEL_NAME) descriptions = ["gentle lo-fi beats"] print("model obtained. generating wav files...")