# ~/docker/ai-image-gen/docker-compose.yml
#
# ═══════════════════════════════════════════════════════════════════════════════
# USAGE
# ═══════════════════════════════════════════════════════════════════════════════
#
# Managed by the ai-portal stack — use localai.mydomain.com to switch stacks.
# To start manually:
#   cd ~/docker/ai-image-gen && docker compose up -d
#
# InvokeAI → https://images.mydomain.com
#
# First model install:
#   Open https://images.mydomain.com → Model Manager → search sd-1.5-inpainting → Install
#
# ═══════════════════════════════════════════════════════════════════════════════
# UPDATES — run update.sh manually when you have a specific reason to update
# ═══════════════════════════════════════════════════════════════════════════════
#
#   Good reasons to update:
#     - A bug you're hitting is fixed
#     - An extension you want requires a newer build
#     - A security issue is announced
#
#   Bad reasons to update:
#     - It's been a while
#     - There's a newer version number
#
#   Run:  ./update.sh
#
# ═══════════════════════════════════════════════════════════════════════════════
# ARCHIVING — run archive.sh after first successful boot
# ═══════════════════════════════════════════════════════════════════════════════
#
#   chmod +x archive.sh && ./archive.sh
#   Store archive/ and data/ on external drive for offline restore.
#
# InvokeAI → https://images.mydomain.com
# ═══════════════════════════════════════════════════════════════════════════════

services:

  # ── InvokeAI ─────────────────────────────────────────────────────────────────
  # Best canvas GUI for inpainting — Photoshop-style mask painting
  # Pinned to v6.11.1 — avoids broken latest-tag issues
  # To upgrade: change tag, run update.sh
  invokeai:
    image: ghcr.io/invoke-ai/invokeai:v6.11.1
    pull_policy: missing
    environment:
      - INVOKEAI_ROOT=/invokeai
      - INVOKEAI_ram=8           # GB of system RAM for model cache
      - INVOKEAI_vram=3          # Turing Tensor Cores — give it VRAM to work with
                                 # drop to 0 if you hit OOM errors
      - INVOKEAI_precision=float16
      - TZ=America/New_York
    volumes:
      - ./data/invokeai:/invokeai   # models, outputs, db, config all live here
    ports:
      - "9090:9090"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped