39 lines
1.2 KiB
YAML
39 lines
1.2 KiB
YAML
services:
|
|
stem-separator:
|
|
build: .
|
|
image: mike-ai/bs-roformer-separator:0.47.0
|
|
container_name: mike-ai-stem-separator
|
|
labels:
|
|
com.mike-ai.stem-separator: "bs-roformer"
|
|
environment:
|
|
NVIDIA_VISIBLE_DEVICES: ${SEPARATOR_GPU_UUID:?set SEPARATOR_GPU_UUID to the RTX 5080 UUID}
|
|
MODEL_FILENAME: model_bs_roformer_ep_317_sdr_12.9755.ckpt
|
|
MODEL_DIR: /models
|
|
JOB_DIR: /data/jobs
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
device_ids: ["${SEPARATOR_GPU_UUID:?set SEPARATOR_GPU_UUID to the RTX 5080 UUID}"]
|
|
capabilities: [gpu]
|
|
ports:
|
|
- "127.0.0.1:${SEPARATOR_PORT:-8007}:8080"
|
|
volumes:
|
|
- ${SEPARATOR_MODEL_DIR:-/data/models/audio-separator}:/models
|
|
- ${SEPARATOR_DATA_DIR:-/data/audio/separation}:/data
|
|
shm_size: "2gb"
|
|
restart: "no"
|
|
healthcheck:
|
|
test: ["CMD-SHELL", "curl -fsS http://127.0.0.1:8080/health | grep -q '\"status\":\"ok\"'"]
|
|
interval: 15s
|
|
timeout: 5s
|
|
start_period: 600s
|
|
retries: 3
|
|
networks: [frontend]
|
|
|
|
networks:
|
|
frontend:
|
|
external: true
|
|
name: mike-ai_frontend
|