Run MusicGen-stereo on Google Colab (Free-tier)

# install packages !python3 -m pip install -U git+https://github.com/facebookresearch/audiocraft#egg=audiocraft  # import packages from audiocraft.models import musicgen from audiocraft.utils.notebook import display_audio import torch # generate music # In this case, we'll try to generate a couple of music prompts = ['futuristic rock music', 'futuristic jazz music'] # describe the music you want to generate  model = musicgen.MusicGen.get_pretrained('facebook/musicgen-stereo-medium') model.set_generation_params(duration=60) res = model.generate(prompts, progress=True) display_audio(res, 32000) 

from audiocraft.data.audio import audio_write for idx, one_wav in enumerate(res): audio_write(f'generated_music_{idx}', one_wav.cpu(), model.sample_rate, strategy="loudness")