mlx-examples/musicgen/README.md

MusicGen

An example of Meta's MusicGen model in MLX.¹ MusicGen is used to generate music from text descriptions.

Setup

Install the requirements:

pip install -r requirements.txt

Example

An example using the model:

import mlx.core as mx
from music_gen import MusicGen
from utils import save_audio

model = MusicGen.from_pretrained("facebook/musicgen-medium")

audio = model.generate("happy rock")

save_audio("out.wav", audio, model.sampling_rate)

---

1. Refer to the arXiv paper and code for more details. ↩︎