feat: change model to musicgen

This commit is contained in:
2024-07-20 14:46:25 +01:00
parent f1ecddc918
commit ec71c0b235

View File

@@ -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...")