services:
   comfyui:
    container_name: ai-comfyui
    build:
      context: .
      dockerfile: dockerfile-comfyui
      args:
        ComfyUI_Release: "0.3.43" # Must be (Maj.Min.Patch) format. -> https://github.com/comfyanonymous/ComfyUI/releases
        Python_Release: "3.12.11" # Must be (Maj.Min.Patch) format. -> https://www.python.org/downloads/source/
        Pytorch_Release: "2.7.1" # Must be (Maj.Min.Patch) format. -> # https://github.com/pytorch/pytorch/releases
        Pytorch_Jobs: 12 # Set the number of jobs for the Pytorch_Build -> (Gb of RAM)/17 rounded down to the nearest whole number
        CUDA_ARCH_LIST: "8.9" # Set the CUDA arch to compile for -> get the current cuda version from 'nvidia-smi'
    volumes:
      - ./data/comfyui/models:/app/comfyui/models
      - ./data/comfyui/output:/app/comfyui/output
    ports:
      - 8188:8188
    restart: always
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]