Prepare isolated YuE2 music evaluation
This commit is contained in:
1 parent
6bd6c48a95
commit
1904f2104f
9 files changed
+102
-263
No files matched your search
@@ -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"]
|
||||
@@ -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.
|
||||
@@ -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
|
||||
}
|
||||
Reference in new issue
Block a user