version: "3.8"

services:
  comfyui:
    image: yanwk/comfyui-boot:latest
    container_name: comfyui
    restart: unless-stopped
    ports:
      - "${COMFYUI_PORT:-8188}:8188"
    volumes:
      - comfyui_data:/root
      - comfyui_models:/root/ComfyUI/models
      - comfyui_output:/root/ComfyUI/output
      - comfyui_input:/root/ComfyUI/input
      - comfyui_custom_nodes:/root/ComfyUI/custom_nodes
    environment:
      - CLI_ARGS=${CLI_ARGS:---listen 0.0.0.0 --port 8188}
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

volumes:
  comfyui_data:
  comfyui_models:
  comfyui_output:
  comfyui_input:
  comfyui_custom_nodes: