24 lines
688 B
YAML
24 lines
688 B
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]
|