version: '3.8'

# Docker Compose override for AMD GPU support (ROCm)
# Usage: docker-compose -f docker-compose.yml -f docker-compose.amd.yml up

services:
  talk2me:
    environment:
      - HSA_OVERRIDE_GFX_VERSION=10.3.0  # Adjust based on your GPU model
      - ROCR_VISIBLE_DEVICES=0  # Use first GPU
    volumes:
      - /dev/kfd:/dev/kfd  # ROCm KFD interface
      - /dev/dri:/dev/dri  # Direct Rendering Interface
    devices:
      - /dev/kfd
      - /dev/dri
    group_add:
      - video  # Required for GPU access
      - render # Required for GPU access