version: '3.8'

services:
  comfyui-amd:
    container_name: comfyui-amd
    build:
      context: ./services/amd
    profiles: ["comfyui-amd"]
    ports:
      - "8188:8188"
    volumes:
      - "./models:/ComfyUI/models"
      - "./output:/ComfyUI/output"
      - "./user:/ComfyUI/user"
    environment:
      - CLI_ARGS=--disable-cuda-malloc --disable-smart-memory
      - HSA_OVERRIDE_GFX_VERSION=10.3.0 # For AMD Radeon RX 6650 XT 
      - MIOPEN_FIND_MODE=FAST
    devices:
      - /dev/dri/renderD128:/dev/dri/renderD128
      - /dev/dri/card0:/dev/dri/card0
      - /dev/kfd:/dev/kfd
    restart: unless-stopped

  comfyui-intel:
    container_name: comfyui-intel
    build:
      context: ./services/intel
    profiles: ["comfyui-intel"]
    ports:
      - "8188:8188"
    volumes:
      - "./models:/ComfyUI/models"
      - "./output:/ComfyUI/output"
      - "./user:/ComfyUI/user"
    environment:
      - CLI_ARGS=--disable-ipex-optimize --use-pytorch-cross-attention --reserve-vram 9.5
      - MIOPEN_FIND_MODE=2
      - MIGRAPHX_MLIR_USE_SPECIFIC_OPS="attention"
    devices:
      - /dev/dri/renderD128:/dev/dri/renderD128
      - /dev/dri/card0:/dev/dri/card0
    restart: unless-stopped

  comfyui-nvidia:
    container_name: comfyui-nvidia
    build:
      context: ./services/nvidia
    profiles: ["comfyui-nvidia"]
    ports:
      - "8188:8188"
    volumes:
      - "./models:/ComfyUI/models"
      - "./output:/ComfyUI/output"
      - "./user:/ComfyUI/user"
    environment:
      - CLI_ARGS=""
    restart: unless-stopped
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities:
                - gpu
                - compute
                - utility