services:
  chatterbox-whisper:
    image: git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
    build:
      context: .
      dockerfile: Dockerfile.rocm
    restart: unless-stopped
    ports:
      - "${WYOMING_PORT:-10200}:10200"
    devices:
      - /dev/kfd
      - /dev/dri
    group_add:
      - video
      - render
    ipc: host
    shm_size: 8g
    security_opt:
      - seccomp=unconfined
    ulimits:
      nofile:
        soft: 65536
        hard: 65536
    volumes:
      - ./config.yaml:/app/config.yaml
      - ./voices:/app/voices
      - ./reference_audio:/app/reference_audio
      - hf_cache:/app/hf_cache
    environment:
      - HF_HUB_ENABLE_HF_TRANSFER=1
      # Required for RX 6700 XT (gfx1031) - not natively supported in ROCm 7.2.
      - HSA_OVERRIDE_GFX_VERSION=10.3.0
      # Disable MIOpen's SQLite cache — avoids crashes writing benchmark results.
      # PyTorch's in-memory benchmark cache still applies within a container run.
      - MIOPEN_DISABLE_CACHE=1
      # Disable MLIR-based ImplicitGEMM solvers. These compile MLIR kernels on the
      # fly and hit 'too many open files' during the exhaustive benchmark search.
      - MIOPEN_DEBUG_CONV_IMPLICIT_GEMM=0
      # - HF_TOKEN=your_token_here

volumes:
  hf_cache: