# Optional one-shot on DGX Spark — runtime image from GHCR, weights on host volume.
#
#   docker compose -f docker/docker-compose.yml pull
#   docker compose -f docker/docker-compose.yml up
#
# First-time with empty models dir (downloads ~91 GB into the volume):
#   DOWNLOAD_WEIGHTS=1 docker compose -f docker/docker-compose.yml up

services:
  comfy:
    image: ghcr.io/drowzeys/keys-heretic-minimax-h3-sol-engine-spark:latest
    build:
      context: ..
      dockerfile: docker/Dockerfile
    network_mode: host
    gpus: all
    environment:
      - DOWNLOAD_WEIGHTS=${DOWNLOAD_WEIGHTS:-0}
      - HF_TOKEN=${HF_TOKEN:-}
      - HF_XET_HIGH_PERFORMANCE=1
    volumes:
      # persist weights / IO on host (default: ~/comfy/ComfyUI/...)
      - ${COMFY_MODELS:-${HOME}/comfy/ComfyUI/models}:/opt/comfy/ComfyUI/models
      - ${COMFY_OUTPUT:-${HOME}/comfy/ComfyUI/output}:/opt/comfy/ComfyUI/output
      - ${COMFY_INPUT:-${HOME}/comfy/ComfyUI/input}:/opt/comfy/ComfyUI/input
    restart: unless-stopped
    # never add --use-sage-attention
    command:
      - python3
      - main.py
      - --listen
      - 0.0.0.0
      - --port
      - "8188"
      - --reserve-vram
      - "8"