Integrate YuE2 as an exclusive Athena mode
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user