services:
  comfyui:
    build:
      context: .
      dockerfile: Dockerfile
    image: aeon-h3-comfy:gb10-h3-lab
    container_name: aeon-h3-comfy-gb10-h3-lab
    runtime: nvidia
    ports:
      - "8202:8188"
    environment:
      SKIP_MODEL_DOWNLOAD: "1"
      SKIP_ABLITERATED: "1"
      COMFYUI_FLAGS: >-
        --listen 0.0.0.0
        --port 8188
        --use-sage-attention
        --disable-pinned-memory
        --reserve-vram 2.0
        --preview-method auto
        --enable-cors-header
        --enable-manager
        --enable-assets
        --extra-model-paths-config /etc/comfyui/h3-ltx-model-paths.yaml
    volumes:
      - ./workspace:/workspace/ComfyUI
      - /home/daniel/StoryStudioAssets/H3-output:/workspace/ComfyUI/output
      - ./h3-ltx-model-paths.yaml:/etc/comfyui/h3-ltx-model-paths.yaml:ro
      - /home/daniel/aeon-spark-test/h3/comfy-models:/h3-models:ro
      - /home/daniel/aeon-spark-test/h3/models/MiniMax-H3-Turbo-FL2VA:/turbo:ro
      - /home/daniel/comfy-data/models/vae:/ltx-video-vae:ro
    shm_size: "32gb"
    ipc: host
    ulimits:
      memlock: -1
      stack: 67108864

  comfyui031:
    build:
      context: .
      dockerfile: Dockerfile.031
    image: aeon-h3-comfy:gb10-h3-lab-031-capture
    container_name: aeon-h3-comfy-gb10-h3-lab-031
    runtime: nvidia
    ports:
      - "8203:8188"
    environment:
      SKIP_MODEL_DOWNLOAD: "1"
      SKIP_ABLITERATED: "1"
      H3_CAPTURE_DIR: /h3-capture
      H3_SAMPLER_CAPTURE_DIR: /h3-sampler-capture
      COMFYUI_FLAGS: >-
        --listen 0.0.0.0
        --port 8188
        --use-sage-attention
        --disable-pinned-memory
        --reserve-vram 2.0
        --preview-method auto
        --enable-cors-header
        --enable-manager
        --enable-assets
        --database-url sqlite:////workspace/ComfyUI/user/comfyui-v031.db
        --extra-model-paths-config /etc/comfyui/h3-ltx-model-paths.yaml
    volumes:
      - ./workspace:/workspace/ComfyUI
      - /home/daniel/StoryStudioAssets/H3-output:/workspace/ComfyUI/output
      - ./h3-ltx-model-paths.yaml:/etc/comfyui/h3-ltx-model-paths.yaml:ro
      - /home/daniel/aeon-spark-test/h3/comfy-models:/h3-models:ro
      - /home/daniel/aeon-spark-test/h3/models/MiniMax-H3-Turbo-FL2VA:/turbo:ro
      - /home/daniel/comfy-data/models/vae:/ltx-video-vae:ro
      - /home/daniel/aeon-spark-test/h3/h3-blackwell-runtime/artifacts/capture:/h3-capture
      - /home/daniel/aeon-spark-test/h3/h3-blackwell-runtime/artifacts/fl2va-sampler-reference:/h3-sampler-capture
    shm_size: "32gb"
    ipc: host
    ulimits:
      memlock: -1
      stack: 67108864