kennethnym/infinifi
1
0
Fork 0
Code Issues 2 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

feat: change model to musicgen

Browse Source
This commit is contained in:
Kenneth
2024-07-20 14:46:25 +01:00
parent f1ecddc918
commit ec71c0b235
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
generate.py
View File

@@ -1,12 +1,12 @@
import torchaudio import torchaudio
from audiocraft.models.magnet import MAGNeT from audiocraft.models.musicgen import MusicGen
from audiocraft.data.audio import audio_write from audiocraft.data.audio import audio_write
MODEL_NAME = "facebook/magnet-medium-30secs" MODEL_NAME = "facebook/musicgen-large"
print(f"getting {MODEL_NAME}...") print(f"getting {MODEL_NAME}...")
model = MAGNeT.get_pretrained(MODEL_NAME) model = MusicGen.get_pretrained(MODEL_NAME)
descriptions = ["gentle lo-fi beats"] descriptions = ["gentle lo-fi beats"]
print("model obtained. generating wav files...") print("model obtained. generating wav files...")