# ComfyUI avec ROCm pour RX 7800 XT
# Architecture: gfx1101 (Navi 32)

services:
  comfyui:
    image: yanwk/comfyui-boot:rocm
    container_name: comfyui
    restart: unless-stopped
    ports:
      - "8188:8188"
    volumes:
      - ./models/checkpoints:/home/user/ComfyUI/models/checkpoints:Z
      - ./models/vae:/home/user/ComfyUI/models/vae:Z
      - ./models/loras:/home/user/ComfyUI/models/loras:Z
      - ./output:/home/user/ComfyUI/output:Z
      - ./input:/home/user/ComfyUI/input:Z
      - ./custom_nodes:/home/user/ComfyUI/custom_nodes:Z
      - ./workflows:/home/user/ComfyUI/workflows:Z
    devices:
      - /dev/kfd:/dev/kfd
      - /dev/dri:/dev/dri
    group_add:
      - video
      - render
    environment:
      - HIP_VISIBLE_DEVICES=0
      - HSA_OVERRIDE_GFX_VERSION=11.0.1
      - PYTORCH_HIP_ALLOC_CONF=expandable_segments:True
      - COMFYUI_PORT=8188
      - COMFYUI_EXTRA_ARGS=--listen 0.0.0.0 --port 8188
    security_opt:
      - seccomp:unconfined
    cap_add:
      - SYS_PTRACE
    shm_size: '8gb'
    stdin_open: true
    tty: true
    deploy:
      resources:
        reservations:
          devices:
            - driver: amd
              count: 1
              capabilities: [gpu]

networks:
  default:
    name: effi-network