Integrate YuE2 as an exclusive Athena mode

This commit is contained in:
Mikei386
2026-09-10 23:53:19 +02:00
parent 5c34afa7fa
commit f2052adb78
14 changed files with 278 additions and 15 deletions
+11
View File
@@ -33,6 +33,8 @@ services:
image: mike-ai/yue2:3b-0.1.6
container_name: mike-ai-yue2-playground
restart: "no"
labels:
com.mike-ai.music-worker: yue2
entrypoint: ["python", "/opt/yue2/webui/server.py"]
environment:
NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
@@ -44,6 +46,9 @@ services:
TRANSFORMERS_OFFLINE: "1"
ports:
- "127.0.0.1:8014:8014"
networks:
frontend:
aliases: [yue2-studio]
volumes:
- /data/models/yue2:/opt/yue2/models:ro
- /data/music/yue2:/workspace/runs
@@ -54,6 +59,7 @@ services:
- driver: nvidia
device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
capabilities: [gpu]
healthcheck:
test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8014/api/state', timeout=2)"]
interval: 5s
@@ -91,3 +97,8 @@ services:
- driver: nvidia
device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
capabilities: [gpu]
networks:
frontend:
external: true
name: mike-ai_frontend