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:
      - "7863:7860"
    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