Usage

A song from a style and lyrics, in Python on a Mac.

pip install mlx mlx-lm numpy tiktoken huggingface_hub
hf download milliwattlabs/YuE2-3B-MLX-4bit --local-dir YuE2-3B-MLX-4bit
cd YuE2-3B-MLX-4bit

python run_song.py --weights weights --out out \
  --style "upbeat indie pop, warm female lead vocal, acoustic guitar, light drums" \
  --lyrics "$(cat lyrics.txt)" --seed 11

The song is written to out/song.wav (48 kHz stereo) and the score to out/abc.txt.

Lyrics

Mark sections with [verse] and [chorus].

Score

--cot sets how much of the score comes before the audio: full (melody and chords, the default), melody or off.

Seed

A seed gives a different song than the same seed in the original PyTorch release.