YuE2-3B: Open Music Generation With Editable Scores From MAP
The MAP team releases YuE2-3B, an open music model that beats Suno v5 on WildSongBench with editable melody and chord scores. ComfyUI native support is now on the yue2 branch.
yue2 branch (PR #16250).
Overview
YuE2-3B is the successor to the original YuE lyrics-to-song model. The team positions it as frontier-quality open music generation: on 192 WildSongBench prompts, YuE2 with best-of-8 sampling reaches a SongBench average of 6.9632, compared with 6.8721 for Suno v5, the highest among all evaluated open and proprietary models.
Frontier song quality and text alignment on 192 WildSongBench prompts. YuE2 uses symbolic planning; Bo8 means best-of-8.
The headline feature is symbolic planning with an editable score. Instead of jumping straight from text to audio, YuE2 first writes a plan containing melody and chords (in ABC notation), then renders audio from it. Because the plan is explicit, you can:
- Compose and edit: full melody plus chords, melody-only, or direct generation without a score
- Bring your own score: feed an existing ABC score and have YuE2 sing it
- Edit with an agent: convert musical feedback into score, style, and lyric revisions, then let the model render the next version. The team demonstrates a 14-version edit chain that moves one song from Mandarin pop to English jazz.
Architecture
One AR–NAR Mixture-of-Transformers backbone writes the score and semantic tokens, then generates acoustic latents through flow matching. The VAE turns them into stereo audio.
YuE2 uses an AR–NAR Mixture-of-Transformers backbone. The autoregressive stage plans the score and semantic tokens; the non-autoregressive stage produces acoustic latents via flow matching, and a dedicated VAE decodes them into 48 kHz stereo audio. Both the planning and synthesis APIs are exposed separately, so developers can inspect or replace the symbolic plan.
Example output
The model card ships full-length demos covering original compositions and covers:
| Example | Style | Length |
|---|---|---|
| Cyber Metal | English cyber metal | 5:00 |
| 今晚不眠 | Mandarin funk / nu-disco | 3:24 |
| Passion | English rock | 3:55 |
| Auld Lang Syne | Jazz-funk cover | 3:10 |
| 最炫民族风 | Ballad cover | 4:45 |
| Jingle Bells | Heavy metal cover | 1:09 |
The cover examples show the model reimagining an existing song in a completely different genre, and the agentic editing demo walks through 9 planning steps and 14 rendered versions of a single song.
Running it
YuE2-3B runs locally on a 24 GB NVIDIA GPU (Linux, Python 3.10+) without quantization. The team distributes an inference wheel alongside the 7.3 GB checkpoint on Hugging Face:
python -m pip install huggingface-hub==0.36.2
hf download m-a-p/YuE2-3B yue2_infer-0.1.5-py3-none-any.whl --local-dir .
python -m pip install ./yue2_infer-0.1.5-py3-none-any.whlThe pipeline exposes Hugging Face loading, text guidance (CFG), and separate planning and synthesis APIs. Full notebooks and the cover/editing recipes are in the README.
Availability
There is no native ComfyUI support yet. The long-standing community node ComfyUI_YuE wraps the YuE series in ComfyUI; it has not been updated for YuE2 at the time of writing, so running YuE2-3B today requires the official Python inference package. A hosted demo is available on the project page.
Native ComfyUI support has landed on the yue2 branch
On September 11, ComfyUI core landed native YuE2 support in PR #16250. At the time of writing it lives on a separate yue2 branch rather than master, so you need to check out that branch (commit d87e12ad) to try it. A ready-to-use all-in-one checkpoint (yue2.safetensors, 7.8 GB) is published on the Comfy-Org Hugging Face account, and a test workflow is attached to the pull request.
Early renders from the native branch sound reasonable, but the integration is still day-one fresh:
- YuE2 pins exact versions of PyTorch, Transformers, and other shared packages, so installing the standalone YuE2 requirements can overwrite packages ComfyUI needs. The native branch avoids the multi-venv problem, but check your environment after setup.
- Community node packs such as EmeraldApple-AI/ComfyUI-YuE2 and a ScryptHunter fork that isolates the YuE2 wheel also appeared within a day; they are experimental one-person projects for now.
- YuE2's own VAE only converts audio to and from continuous acoustic latents, which means LoRA training for YuE2 is not feasible yet: the discrete semantic music tokens and a YuE2-specific training pipeline have not been released.
Comments
Sign in with GitHub to join the discussion.