name: comfyui

services:
  comfyui:
    image: ${COMFYUI_IMAGE:-yanwk/comfyui-boot:cu126-slim}
    container_name: comfyui
    profiles:
      - comfyui
    restart: unless-stopped
    init: true
    ports:
      - "${COMFYUI_PORT:-8188}:8188"
    environment:
      - CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-0}
      - NVIDIA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-0}
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - CLI_ARGS=${CLI_ARGS:-}
    volumes:
      - ${DATA_ROOT:-/data}/apps/comfyui/storage:/root
      - ${DATA_ROOT:-/data}/apps/comfyui/models:/root/ComfyUI/models
      - ${DATA_ROOT:-/data}/apps/comfyui/cache/hf-hub:/root/.cache/huggingface/hub
      - ${DATA_ROOT:-/data}/apps/comfyui/cache/torch-hub:/root/.cache/torch/hub
      - ${DATA_ROOT:-/data}/apps/comfyui/input:/root/ComfyUI/input
      - ${DATA_ROOT:-/data}/apps/comfyui/output:/root/ComfyUI/output
      - ${DATA_ROOT:-/data}/apps/comfyui/custom_nodes:/root/ComfyUI/custom_nodes
      - ${DATA_ROOT:-/data}/apps/comfyui/workflows:/root/ComfyUI/user/default/workflows
    gpus: all
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8188/"]
      interval: 1m
      timeout: 30s
      retries: 5
      start_period: 3m