58 lines
1.7 KiB
YAML
58 lines
1.7 KiB
YAML
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]
|
|
|
|
yue2-ui:
|
|
profiles: ["playground"]
|
|
build:
|
|
context: .
|
|
args:
|
|
YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
|
|
image: mike-ai/yue2:3b-0.1.6
|
|
container_name: mike-ai-yue2-playground
|
|
restart: "no"
|
|
entrypoint: ["python", "/opt/yue2-playground/server.py"]
|
|
environment:
|
|
NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
|
|
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
|
YUE2_UI_HOST: "0.0.0.0"
|
|
YUE2_UI_PORT: "8014"
|
|
ports:
|
|
- "127.0.0.1:8014:8014"
|
|
volumes:
|
|
- /data/models/yue2:/workspace/models:ro
|
|
- /data/music/yue2:/workspace/runs
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- 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/health', timeout=2)"]
|
|
interval: 5s
|
|
timeout: 3s
|
|
retries: 12
|
|
start_period: 5s
|