40 lines
1.0 KiB
YAML
40 lines
1.0 KiB
YAML
services:
|
|
heartmula-test:
|
|
build:
|
|
context: .
|
|
args:
|
|
HEARTLIB_COMMIT: 3783bdb8441f2c298b1e64c8651173aac200361c
|
|
image: mike-ai/heartmula:3b-3783bdb
|
|
container_name: mike-ai-heartmula-test
|
|
restart: "no"
|
|
environment:
|
|
NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe,GPU-4834d9d7-5b61-3004-1fb3-4ae49d482d4b"
|
|
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
|
HEARTMULA_MULA_DEVICE: cuda:0
|
|
HEARTMULA_CODEC_DEVICE: cuda:1
|
|
volumes:
|
|
- /data/models/heartmula:/models:ro
|
|
- /data/music/heartmula:/output
|
|
ports:
|
|
- "127.0.0.1:7863:7860"
|
|
networks:
|
|
- frontend
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: [gpu]
|
|
healthcheck:
|
|
test: ["CMD", "curl", "-fsS", "http://127.0.0.1:7860/healthz"]
|
|
interval: 15s
|
|
timeout: 5s
|
|
retries: 20
|
|
start_period: 180s
|
|
|
|
networks:
|
|
frontend:
|
|
external: true
|
|
name: mike-ai_frontend
|