Prepare isolated YuE2 music evaluation

This commit is contained in:
Mikei386 committed 2026-09-10 21:25:59 +02:00
1 parent 6bd6c48a95
commit 1904f2104f
9 files changed
+102 -263

No files matched your search

+17
View File
@@ -0,0 +1,17 @@
FROM python:3.12-slim-bookworm
ARG YUE2_COMMIT=9c6c4b349be978b06a9d0d958471a07a6cdeff4d
RUN apt-get update \
&& apt-get install -y --no-install-recommends ca-certificates git libsndfile1 \
&& rm -rf /var/lib/apt/lists/*
RUN git clone https://github.com/multimodal-art-projection/YuE.git /opt/yue2 \
&& git -C /opt/yue2 checkout "${YUE2_COMMIT}" \
&& python -m pip install --no-cache-dir /opt/yue2
WORKDIR /workspace
ENV PYTHONUNBUFFERED=1 \
YUE2_KIT=/workspace
ENTRYPOINT ["yue2"]
+52
View File
@@ -0,0 +1,52 @@
# YuE2 3B isolated quality test
Prepared, non-starting evaluation of `m-a-p/YuE2-3B` with the standard
`m-a-p/YuE2-Vae` listening decoder. The source is pinned to the official
`yue2-v0.1.6` commit `9c6c4b349be978b06a9d0d958471a07a6cdeff4d`.
Preparation on Athena is complete. The model and VAE files were checked
against their published `weights_manifest.json` SHA-256 values. The Docker
image is built, but no YuE2 container has been created or started.
## Safety and isolation
- This experiment is not part of the profile controller or dashboard.
- The Compose service uses the `manual` profile, has no restart policy and
cannot start through an ordinary `docker compose up`.
- Only the RTX 5080 is exposed to the container.
- Building and downloading do not load the model or use a GPU.
- Do not start it while another Athena GPU job is active.
## Persistent files
```text
/data/models/yue2/
├── YuE2-3B/
└── YuE2-Vae/
/data/music/yue2/
```
The initial control request is a true empty-lyrics instrumental request. No
invented `[Instrumental]` lyrics marker is used.
## Manual test (only after GPU availability was checked)
From `/opt/mike-ai/yue2-3b` on Athena:
```sh
docker compose --profile manual run --rm yue2-test generate \
--offline \
--device cuda:0 \
--budget 16 \
--request /workspace/requests/instrumental-synthwave.json \
--output /workspace/runs
```
Start with the official unquantized BF16 path. If and only if this fails from
VRAM pressure, repeat with `--quantization fp8 --offload-ar`; keep the outputs
separate because that is a different inference configuration.
YuE2 is newly released and officially specifies a 24-GB BF16 GPU. Readiness of
this image and the downloaded weights is not evidence that the 16-GB RTX 5080
run will fit or that its audio quality is acceptable.
+23
View File
@@ -0,0 +1,23 @@
services:
yue2-test:
profiles: ["manual"]
build:
context: .
args:
YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
image: mike-ai/yue2:3b-0.1.6
restart: "no"
environment:
NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
NVIDIA_DRIVER_CAPABILITIES: compute,utility
volumes:
- /data/models/yue2:/workspace/models:ro
- /data/music/yue2:/workspace/runs
- ./requests:/workspace/requests:ro
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
capabilities: [gpu]
@@ -0,0 +1,7 @@
{
"id": "instrumental_synthwave_control",
"style": "Instrumental, synthwave, synth-pop, energetic, memorable lead melody, arpeggiated synthesizer, analog synthesizer, drum machine, driving bass, 118 BPM, no vocals",
"lyrics": "",
"cot": "full",
"seed": 831001
}