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
    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
      # Set your GPU architecture:
      #   10.3.0 = RX 5000/6000 series
      #   11.0.0 = RX 7000 series
      #   9.0.6  = Vega
      - HSA_OVERRIDE_GFX_VERSION=10.3.0
      # - HF_TOKEN=your_token_here

volumes:
  hf_cache: