Upload 3 files

README.md

@@ -0,0 +1,206 @@
+---
+license: apache-2.0
+tags:
+  - comfyui
+  - runpod
+  - faceswap
+---
+
+# ComfyUI FaceSwap — готовый темплейт для RunPod
+
+Поднимает на RunPod полностью настроенный ComfyUI с пайплайном замены лица:
+
+1. **Flux.2 Klein 9b** — переносит голову с одного фото на другое, сохраняя свет, ракурс и пропорции
+2. **Z-Image Turbo + FaceDetailer** — заново отрисовывает лицо в высоком разрешении, сюда подключается персональная LoRA
+
+Оба этапа работают в связке и оба обязательны. Flux хорошо ставит голову по свету и ракурсу, но лицо у него выходит лишь похожим. Z-Image затем вырезает это лицо и рисует его заново — уже с обученной на конкретном человеке LoRA. Отсюда и настоящее сходство.
+
+Ничего устанавливать руками не нужно. Один раз создаёшь темплейт — дальше запуск в два клика.
+
+**Железо:** проверено на L40S (48 ГБ VRAM). На 24 ГБ тоже должно пойти, но не тестировалось.
+
+---
+
+## Что нужно иметь
+
+- аккаунт на [RunPod](https://runpod.io) с пополненным балансом
+- аккаунт на [HuggingFace](https://huggingface.co) — бесплатный
+
+---
+
+## Шаг 1. Токен HuggingFace
+
+Нужен, чтобы скачать модель Flux. Свой, личный — чужие токены не используй.
+
+1. Открой https://huggingface.co/settings/tokens
+2. **Create new token**
+3. Тип — **Read** (не Write)
+4. Название любое, например `runpod`
+5. **Create token** → скопируй и сохрани в блокнот
+
+HuggingFace показывает токен только один раз. Потеряешь — сделаешь новый, ничего страшного.
+
+---
+
+## Шаг 2. Принять лицензию Flux
+
+1. Открой https://huggingface.co/black-forest-labs/FLUX.2-klein-9b-fp8
+2. Убедись, что залогинен
+3. Нажми **Agree** на серой плашке
+
+Выбирать там нечего, полей нет — просто кнопка.
+
+> ⚠️ У FLUX.2 Klein **некоммерческая лицензия**. Использование результатов в коммерческих целях она не разрешает. Это твоя зона ответственности.
+>
+> Обойти этот шаг и всё равно пользоваться воркфлоу не получится: Flux здесь делает первый этап, без него замена лица не работает. Переключатель `WANT_FLUX=0` из таблицы настроек существует для тех, кто собирает что-то своё только на Z-Image — для этого воркфлоу его выключать нельзя.
+
+---
+
+## Шаг 3. Создать темплейт в RunPod
+
+Консоль RunPod → левое меню **Templates** → **New Template**.
+
+| Поле | Значение |
+|---|---|
+| Template Name | `ComfyUI-FaceSwap` |
+| Template Type | `Pod` |
+| Container Image | `runpod/pytorch:1.3.1-cu1281-torch291-ubuntu2404` |
+| Container Disk | `20` GB |
+| Volume Disk | `80` GB |
+| Volume Mount Path | `/workspace` |
+| Expose HTTP Ports | `8188,8888` |
+| Expose TCP Ports | `22` |
+
+**Container Start Command** (в некоторых версиях поле называется Docker Command):
+
+```
+bash -c "curl -sL https://huggingface.co/cocount21/comfy-faceswap-setup/resolve/main/provision.sh -o /provision.sh && bash /provision.sh 2>&1 | tee /workspace/provision.log"
+```
+
+**Environment Variables** — добавь одну переменную:
+
+| Key | Value |
+|---|---|
+| `HF_TOKEN` | **оставь пустым** |
+
+Пустым — потому что заполнять её ты будешь при запуске пода, своим токеном. Так она видна в форме деплоя и её не забудешь.
+
+Нажми **Save Template**.
+
+---
+
+## Шаг 4. Запустить
+
+1. **Pods** → **Deploy**
+2. Выбери **L40S**
+3. **Change Template** → свой `ComfyUI-FaceSwap`
+4. В блоке Environment Variables впиши **свой** токен в `HF_TOKEN`
+5. **Deploy On-Demand**
+
+---
+
+## Шаг 5. Подождать
+
+Первый запуск — 10–20 минут, качается около 40 ГБ моделей. ComfyUI стартует последним, раньше времени его дёргать бесполезно.
+
+Смотреть прогресс: карточка пода → **Logs** → **Container Logs**.
+
+В конце скрипт печатает отчёт:
+
+```
+===========================  SUMMARY  ===========================
+  OK       Flux-2-Klein-9b-fp8.safetensors
+  OK       qwen_3_8b_fp8mixed.safetensors
+  ...
+=================================================================
+  time: 812 s
+
+  ALL GOOD - ComfyUI is starting on port 8188
+```
+
+Зелёное `ALL GOOD` — всё готово. Красные строки — смотри раздел «Если что-то пошло не так».
+
+---
+
+## Шаг 6. Работать
+
+1. Карточка пода → **Connect** → **HTTP Service → :8188**
+2. Воркфлоу уже загружен — в левой панели ComfyUI, раздел Workflows
+3. Загрузи две картинки:
+   - группа **Your-Image** — фото-основа: тело, поза, фон
+   - группа **Reference-Image** — лицо, которое переносим
+4. **Run**
+
+Результат — в ноде **SaveImage** с жёлтой рамкой. Вторая SaveImage показывает промежуточный результат Flux, до починки лица.
+
+### Своя LoRA
+
+Залей свою LoRA в любой публичный репозиторий на HuggingFace и при деплое добавь переменную:
+
+```
+EXTRA_LORAS = твой_ник/твой_репозиторий:имя_файла.safetensors
+```
+
+Несколько — через запятую. Файлы лягут в `models/loras/`, останется выбрать их в ноде.
+
+---
+
+## Настройки (Environment Variables)
+
+| Переменная | По умолчанию | Что делает |
+|---|---|---|
+| `HF_TOKEN` | пусто | твой read-токен HuggingFace |
+| `WANT_FLUX` | `1` | этап Flux.2 Klein, ~19 ГБ, gated, non-commercial. **Для этого воркфлоу выключать нельзя** |
+| `WANT_ZIMAGE` | `1` | этап Z-Image Turbo, ~21 ГБ, Apache 2.0. **Для этого воркфлоу выключать нельзя** |
+| `WANT_DETECTORS` | `1` | детекторы лица для FaceDetailer, ~0.4 ГБ. **Выключать нельзя** |
+| `EXTRA_LORAS` | пусто | свои LoRA: `repo:file.safetensors`, через запятую |
+| `GET_WORKFLOWS` | `1` | подтянуть воркфлоу автоматически |
+| `START_JUPYTER` | `0` | поднять JupyterLab на порту 8888 |
+| `SKIP_MODELS` | `0` | только ноды, без моделей — для быстрой проверки |
+| `COMFY_PORT` | `8188` | порт ComfyUI |
+| `COMFY_REF` | пусто | закрепить ComfyUI на конкретном коммите |
+
+---
+
+## Если что-то пошло не так
+
+Скрипт проверяет доступ к Flux **до** скачивания и объясняет проблему прямо в логе.
+
+| В логе | Причина | Решение |
+|---|---|---|
+| `HTTP 401` | токен неверный или пустой | сделать Read-токен, вписать в `HF_TOKEN` |
+| `HTTP 403` | лицензия не принята | нажать Agree на странице модели |
+| `BAD SIZE` | файл скачался битым | перезапустить под, скрипт перекачает сам |
+| `TORCH CHANGED` | нода подменила torch | сообщи автору темплейта |
+| `NODE MISSING` | не склонировалась нода | перезапустить под |
+
+Полный лог лежит в поде: `/workspace/provision.log`
+
+---
+
+## Что внутри
+
+**Ноды:** ComfyUI-Manager, Impact-Pack, Impact-Subpack, Easy-Use, rgthree-comfy, utils-nodes
+
+**Модели:**
+
+| Файл | Размер | Лицензия |
+|---|---|---|
+| Flux-2-Klein-9b-fp8 | 8.8 ГБ | FLUX Non-Commercial |
+| qwen_3_8b_fp8mixed | 8.1 ГБ | — |
+| flux2-vae | 0.3 ГБ | — |
+| bfs_head_v1 (LoRA свапа) | 0.6 ГБ | не указана автором |
+| z_image_turbo_bf16 | 11.5 ГБ | Apache 2.0 |
+| qwen_3_4b | 7.5 ГБ | Apache 2.0 |
+| ae (VAE) | 0.3 ГБ | Apache 2.0 |
+| face_yolov8m + SAM | 0.4 ГБ | — |
+
+Скрипт сверяет размер каждого скачанного файла с эталоном. Битый файл будет пойман сразу, а не через час непонятных ошибок.
+
+---
+
+## Безопасность
+
+- Токен вписывай **свой**. Переменные окружения видны изнутри пода любому, у кого есть к нему доступ.
+- Используй **Read**-токен, не Write. Read не даёт ничего испортить.
+- Не вставляй токен в сам скрипт — только в переменную окружения.

provision.sh

@@ -0,0 +1,388 @@
+#!/usr/bin/env bash
+# ===========================================================================
+#  ComfyUI provisioning for RunPod
+#  Flux.2 Klein head-swap  +  Z-Image Turbo face restore
+# ===========================================================================
+#
+#  RunPod template -> "Container Start Command":
+#    bash -c "curl -sL https://huggingface.co/cocount21/comfy-faceswap-setup/resolve/main/provision.sh -o /provision.sh && bash /provision.sh 2>&1 | tee /workspace/provision.log"
+#
+#  ------------------------------- SETTINGS -------------------------------
+#  HF_TOKEN        your OWN HuggingFace read token.
+#                  Needed only when WANT_FLUX=1 (Klein 9b is a gated repo).
+#                  Leave it EMPTY in the shared template - every user
+#                  types in their own at deploy time.
+#
+#  WANT_FLUX=1     Flux.2 Klein 9b stage        (~19 GB, gated, non-commercial)
+#  WANT_ZIMAGE=1   Z-Image Turbo stage          (~21 GB, Apache-2.0, open)
+#  WANT_DETECTORS=1  face detector + SAM        (~0.4 GB, needed by FaceDetailer)
+#
+#  EXTRA_LORAS=""  your own LoRAs, comma separated, "repo:path-in-repo"
+#                  e.g.  EXTRA_LORAS="myname/myrepo:my_lora.safetensors"
+#
+#  GET_WORKFLOWS=1 auto-install the .json workflows into ComfyUI
+#  START_JUPYTER=0 also run JupyterLab on :8888
+#  SKIP_MODELS=0   set up nodes only, download nothing
+#  COMFY_PORT=8188
+#  COMFY_REF=""    pin ComfyUI to a git commit/tag instead of latest
+# ===========================================================================
+set -uo pipefail
+
+ROOT="${WORKSPACE:-/workspace}"
+COMFY="$ROOT/ComfyUI"
+STAGE="$ROOT/.dl"
+CONS="$ROOT/.torch-constraints.txt"
+M="$COMFY/models"
+PORT="${COMFY_PORT:-8188}"
+
+WANT_FLUX="${WANT_FLUX:-1}"
+WANT_ZIMAGE="${WANT_ZIMAGE:-1}"
+WANT_DETECTORS="${WANT_DETECTORS:-1}"
+EXTRA_LORAS="${EXTRA_LORAS:-}"
+GET_WORKFLOWS="${GET_WORKFLOWS:-1}"
+SETUP_REPO="${SETUP_REPO:-cocount21/comfy-faceswap-setup}"
+
+GATED_REPO="black-forest-labs/FLUX.2-klein-9b-fp8"
+GATED_FILE="flux-2-klein-9b-fp8.safetensors"
+
+REPORT=()
+PROBLEMS=0
+
+log()  { printf '\n\033[1;36m==> %s\033[0m\n' "$*"; }
+ok()   { printf '    \033[1;32m%s\033[0m\n' "$*"; }
+warn() { printf '    \033[1;33m[warn] %s\033[0m\n' "$*"; }
+err()  { printf '    \033[1;31m[ERROR] %s\033[0m\n' "$*"; PROBLEMS=$((PROBLEMS+1)); }
+die()  { printf '\n\033[1;31m[fail] %s\033[0m\n' "$*"; exit 1; }
+banner() {
+    printf '\n\033[1;31m'
+    printf '#%.0s' {1..74}; printf '\n'
+    while IFS= read -r l; do printf '#  %-69s#\n' "$l"; done <<< "$1"
+    printf '#%.0s' {1..74}; printf '\033[0m\n\n'
+}
+
+mkdir -p "$ROOT"; cd "$ROOT"
+START_TS=$(date +%s)
+
+# --------------------------------------------------------------- ssh -------
+if [ -n "${PUBLIC_KEY:-}" ]; then
+    mkdir -p ~/.ssh && chmod 700 ~/.ssh
+    echo "$PUBLIC_KEY" >> ~/.ssh/authorized_keys && chmod 600 ~/.ssh/authorized_keys
+    ssh-keygen -A >/dev/null 2>&1
+    /usr/sbin/sshd >/dev/null 2>&1 && ok "sshd up"
+fi
+
+# ==========================================================================
+#  PREFLIGHT - check the gated repo BEFORE downloading tens of gigabytes
+# ==========================================================================
+if [ "${SKIP_MODELS:-0}" != "1" ] && [ "$WANT_FLUX" = "1" ]; then
+    log "Preflight: access to the gated Flux repo"
+    CODE=$(curl -s -o /dev/null -w '%{http_code}' -L -r 0-0 \
+           -H "Authorization: Bearer ${HF_TOKEN:-nope}" \
+           "https://huggingface.co/$GATED_REPO/resolve/main/$GATED_FILE" 2>/dev/null)
+    case "$CODE" in
+        200|206)
+            ok "access confirmed (HTTP $CODE)" ;;
+        401)
+            banner "NO ACCESS TO THE FLUX MODEL  (HTTP 401 - bad or missing token)
+
+Fix, takes 2 minutes:
+
+ 1. Create a HuggingFace account:  https://huggingface.co/join
+ 2. Make a READ token:            https://huggingface.co/settings/tokens
+      Create new token -> type Read -> copy it
+ 3. Put it into this pod's HF_TOKEN environment variable and restart.
+
+Skipping the Flux stage for now - Z-Image will still be installed."
+            WANT_FLUX=0; PROBLEMS=$((PROBLEMS+1)) ;;
+        403)
+            banner "FLUX LICENCE NOT ACCEPTED  (HTTP 403)
+
+Your token works, but you have not accepted the model licence yet.
+
+ 1. Open (logged in):
+      https://huggingface.co/black-forest-labs/FLUX.2-klein-9b-fp8
+ 2. Press the  Agree  button on the grey banner.
+ 3. Restart this pod.
+
+Note: FLUX.2 Klein is under a NON-COMMERCIAL licence.
+
+Skipping the Flux stage for now - Z-Image will still be installed."
+            WANT_FLUX=0; PROBLEMS=$((PROBLEMS+1)) ;;
+        *)
+            warn "unexpected HTTP $CODE - will try anyway" ;;
+    esac
+fi
+
+# ------------------------------------------------------------ apt deps -----
+if [ ! -f "$ROOT/.apt-done" ]; then
+    log "System packages"
+    export DEBIAN_FRONTEND=noninteractive
+    apt-get update -qq >/dev/null 2>&1
+    apt-get install -y -qq git aria2 libgl1 libglib2.0-0t64 >/dev/null 2>&1 \
+      || apt-get install -y -qq git aria2 libgl1 libglib2.0-0 >/dev/null 2>&1 \
+      || warn "apt partially failed (opencv-based nodes may complain)"
+    touch "$ROOT/.apt-done"; ok "done"
+fi
+
+# ------------------------------------------------- pin torch, get hf cli ---
+log "Python tooling"
+python - > "$CONS" <<'PYEOF'
+import torch
+print("torch==" + torch.__version__)
+for m in ("torchvision", "torchaudio"):
+    try:
+        print(m + "==" + __import__(m).__version__)
+    except Exception:
+        pass
+PYEOF
+ok "torch pinned: $(head -1 "$CONS")"
+
+pip install -q -U pip >/dev/null 2>&1
+pip install -q -U "huggingface_hub[hf_transfer,cli]" >/dev/null 2>&1
+export HF_HUB_ENABLE_HF_TRANSFER=1
+HF_BIN="$(command -v hf || command -v huggingface-cli || true)"
+[ -n "$HF_BIN" ] || die "huggingface cli not found"
+ok "hf cli ready"
+
+# -------------------------------------------------------------- ComfyUI ---
+if [ ! -d "$COMFY/.git" ]; then
+    log "Cloning ComfyUI"
+    if [ -n "${COMFY_REF:-}" ]; then
+        git clone https://github.com/comfyanonymous/ComfyUI "$COMFY" >/dev/null 2>&1 \
+            && git -C "$COMFY" checkout "$COMFY_REF" >/dev/null 2>&1 \
+            && ok "pinned to $COMFY_REF" || die "clone/checkout failed"
+    else
+        git clone --depth 1 https://github.com/comfyanonymous/ComfyUI "$COMFY" >/dev/null 2>&1 \
+            || die "clone failed"
+    fi
+fi
+log "ComfyUI requirements"
+pip install -q -r "$COMFY/requirements.txt" -c "$CONS" >/dev/null 2>&1 || warn "some reqs failed"
+ok "done"
+
+# --------------------------------------------------------- custom nodes ---
+NODES=(
+  "https://github.com/ltdrdata/ComfyUI-Manager"
+  "https://github.com/ltdrdata/ComfyUI-Impact-Pack"
+  "https://github.com/ltdrdata/ComfyUI-Impact-Subpack"
+  "https://github.com/yolain/ComfyUI-Easy-Use"
+  "https://github.com/rgthree/rgthree-comfy"
+  "https://github.com/zhangp365/ComfyUI-utils-nodes"
+)
+log "Custom nodes"
+mkdir -p "$COMFY/custom_nodes"
+for url in "${NODES[@]}"; do
+    name="$(basename "$url")"
+    dir="$COMFY/custom_nodes/$name"
+    if [ ! -d "$dir/.git" ]; then
+        if git clone --depth 1 "$url" "$dir" >/dev/null 2>&1; then
+            ok "cloned  $name"
+        else
+            err "clone failed: $name"; REPORT+=("NODE MISSING  $name"); continue
+        fi
+    else
+        ok "present $name"
+    fi
+    if [ -f "$dir/requirements.txt" ]; then
+        pip install -q -r "$dir/requirements.txt" -c "$CONS" >/dev/null 2>&1 || warn "pip failed: $name"
+    fi
+    if [ -f "$dir/install.py" ]; then
+        ( cd "$dir" && python install.py >/dev/null 2>&1 ) || warn "install.py failed: $name"
+    fi
+done
+
+TORCH_NOW="$(python -c 'import torch; print(torch.__version__)' 2>/dev/null || echo BROKEN)"
+if [ "torch==$TORCH_NOW" = "$(head -1 "$CONS")" ]; then
+    ok "torch intact ($TORCH_NOW)"
+else
+    err "TORCH CHANGED -> $TORCH_NOW (expected $(head -1 "$CONS"))"
+    REPORT+=("TORCH was replaced by a custom node - generation may fail")
+fi
+
+# ============================================================== models =====
+mkdir -p "$M/diffusion_models" "$M/text_encoders" "$M/vae" "$M/loras" "$M/sams" "$M/ultralytics/bbox"
+
+hf_size() {   # repo  path  ->  expected bytes (0 if unknown)
+    python - "$1" "$2" <<'PYEOF'
+import sys, os, json, urllib.request
+repo, path = sys.argv[1], sys.argv[2]
+try:
+    req = urllib.request.Request(
+        "https://huggingface.co/api/models/%s/tree/main?recursive=true" % repo)
+    tok = os.environ.get("HF_TOKEN")
+    if tok:
+        req.add_header("Authorization", "Bearer " + tok)
+    for e in json.load(urllib.request.urlopen(req, timeout=30)):
+        if e.get("path") == path and e.get("type") == "file":
+            print((e.get("lfs") or {}).get("size") or e.get("size") or 0)
+            break
+    else:
+        print(0)
+except Exception:
+    print(0)
+PYEOF
+}
+
+fetch() {   # repo  path-in-repo  dest  expected-bytes|auto
+    local repo="$1" path="$2" dest="$3" want="$4" have base
+    base="$(basename "$dest")"
+    [ "$want" = "auto" ] && want="$(hf_size "$repo" "$path")"
+    mkdir -p "$(dirname "$dest")"
+    if [ -f "$dest" ]; then
+        have=$(stat -c %s "$dest")
+        if [ "$want" = "0" ] || [ "$have" = "$want" ]; then
+            ok "cached  $base"; REPORT+=("OK       $base"); return 0
+        fi
+        warn "size mismatch ($have != $want) -> refetch $base"
+        rm -f "$dest"
+    fi
+    printf '    downloading %s ...\n' "$base"
+    rm -rf "$STAGE"
+    if ! "$HF_BIN" download "$repo" "$path" --local-dir "$STAGE" >/dev/null 2>&1; then
+        err "download failed: $repo :: $path"; REPORT+=("MISSING  $base")
+        rm -rf "$STAGE"; return 1
+    fi
+    mv -f "$STAGE/$path" "$dest"; rm -rf "$STAGE"
+    have=$(stat -c %s "$dest")
+    if [ "$want" = "0" ] || [ "$have" = "$want" ]; then
+        ok "ok      $base  ($have B)"; REPORT+=("OK       $base")
+    else
+        err "BAD SIZE $base: $have != $want"; REPORT+=("CORRUPT  $base")
+        return 1
+    fi
+}
+
+fetch_url() {   # url  dest  expected-bytes
+    local url="$1" dest="$2" want="$3" have base
+    base="$(basename "$dest")"
+    mkdir -p "$(dirname "$dest")"
+    if [ -f "$dest" ] && [ "$(stat -c %s "$dest")" = "$want" ]; then
+        ok "cached  $base"; REPORT+=("OK       $base"); return 0
+    fi
+    rm -f "$dest"
+    printf '    downloading %s ...\n' "$base"
+    aria2c -x8 -s8 -q --allow-overwrite=true \
+           -d "$(dirname "$dest")" -o "$base" "$url" >/dev/null 2>&1 \
+        || curl -sL "$url" -o "$dest" \
+        || { err "download failed: $url"; REPORT+=("MISSING  $base"); return 1; }
+    have=$(stat -c %s "$dest")
+    if [ "$have" = "$want" ]; then
+        ok "ok      $base  ($have B)"; REPORT+=("OK       $base")
+    else
+        err "BAD SIZE $base: $have != $want"; REPORT+=("CORRUPT  $base")
+    fi
+}
+
+if [ "${SKIP_MODELS:-0}" = "1" ]; then
+    log "Models: skipped (SKIP_MODELS=1)"
+else
+    if [ "$WANT_FLUX" = "1" ]; then
+        log "Models: Flux.2 Klein 9b  (~19 GB)"
+        fetch "$GATED_REPO" "$GATED_FILE" \
+              "$M/diffusion_models/Flux-2-Klein-9b-fp8.safetensors" 9433061528
+        fetch Comfy-Org/vae-text-encorder-for-flux-klein-9b \
+              split_files/text_encoders/qwen_3_8b_fp8mixed.safetensors \
+              "$M/text_encoders/qwen_3_8b_fp8mixed.safetensors" 8664848742
+        fetch Comfy-Org/vae-text-encorder-for-flux-klein-9b \
+              split_files/vae/flux2-vae.safetensors \
+              "$M/vae/flux2-vae.safetensors" 336211292
+        fetch bond12321/bfs-head-klein9b \
+              bfs_head_v1_flux-klein_9b_step3500_rank128.safetensors \
+              "$M/loras/bfs_head_v1_flux-klein_9b_step3500_rank128.safetensors" 662729912
+    else
+        log "Models: Flux stage skipped"
+        REPORT+=("SKIPPED  Flux.2 Klein stage")
+    fi
+
+    if [ "$WANT_ZIMAGE" = "1" ]; then
+        log "Models: Z-Image Turbo  (~21 GB)"
+        fetch Comfy-Org/z_image_turbo \
+              split_files/diffusion_models/z_image_turbo_bf16.safetensors \
+              "$M/diffusion_models/z_image_turbo_bf16.safetensors" 12309866400
+        fetch Comfy-Org/z_image_turbo \
+              split_files/text_encoders/qwen_3_4b.safetensors \
+              "$M/text_encoders/qwen_3_4b.safetensors" 8044982048
+        fetch Comfy-Org/z_image_turbo \
+              split_files/vae/ae.safetensors \
+              "$M/vae/ae.safetensors" 335304388
+    else
+        log "Models: Z-Image stage skipped"
+        REPORT+=("SKIPPED  Z-Image stage")
+    fi
+
+    if [ "$WANT_DETECTORS" = "1" ]; then
+        log "Models: face detectors"
+        fetch Bingsu/adetailer face_yolov8m.pt \
+              "$M/ultralytics/bbox/face_yolov8m.pt" 52026019
+        fetch_url https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth \
+              "$M/sams/sam_vit_b_01ec64.pth" 375042383
+    fi
+
+    # -- user supplied LoRAs -------------------------------------------------
+    if [ -n "$EXTRA_LORAS" ]; then
+        log "Models: your own LoRAs"
+        IFS=',' read -ra ITEMS <<< "$EXTRA_LORAS"
+        for item in "${ITEMS[@]}"; do
+            item="$(echo "$item" | xargs)"
+            [ -z "$item" ] && continue
+            lrepo="${item%%:*}"; lpath="${item#*:}"
+            if [ "$lrepo" = "$item" ] || [ -z "$lpath" ]; then
+                err "bad EXTRA_LORAS entry: '$item'  (expected repo:file.safetensors)"
+                continue
+            fi
+            fetch "$lrepo" "$lpath" "$M/loras/$(basename "$lpath")" auto
+        done
+    fi
+fi
+
+# ----------------------------------------------------------- workflows ----
+if [ "$GET_WORKFLOWS" = "1" ]; then
+    log "Workflows"
+    WFDIR="$COMFY/user/default/workflows"
+    mkdir -p "$WFDIR"
+    rm -rf "$STAGE"
+    if "$HF_BIN" download "$SETUP_REPO" --include "workflows/*.json" \
+           --local-dir "$STAGE" >/dev/null 2>&1 && [ -d "$STAGE/workflows" ]; then
+        cp -f "$STAGE"/workflows/*.json "$WFDIR"/ 2>/dev/null \
+            && ok "installed: $(ls -1 "$WFDIR" | tr '\n' ' ')" \
+            || warn "no .json found in $SETUP_REPO/workflows/"
+    else
+        warn "no workflows/ folder in $SETUP_REPO - load the .json by hand"
+    fi
+    rm -rf "$STAGE"
+fi
+
+# ------------------------------------------------------------- jupyter ----
+if [ "${START_JUPYTER:-0}" = "1" ] && command -v jupyter >/dev/null 2>&1; then
+    JT="${JUPYTER_PASSWORD:-$(python -c 'import secrets; print(secrets.token_hex(16))')}"
+    nohup jupyter lab --allow-root --no-browser --ip=0.0.0.0 --port=8888 \
+        --ServerApp.token="$JT" --ServerApp.allow_origin='*' \
+        --notebook-dir="$ROOT" >"$ROOT/jupyter.log" 2>&1 &
+    log "JupyterLab :8888   token = $JT"
+fi
+
+# ======================================================== final report ====
+ELAPSED=$(( $(date +%s) - START_TS ))
+printf '\n\n'
+printf '\033[1m===========================  SUMMARY  ===========================\033[0m\n'
+for line in "${REPORT[@]}"; do
+    case "$line" in
+        OK*)      printf '  \033[1;32m%s\033[0m\n' "$line" ;;
+        SKIPPED*) printf '  \033[1;34m%s\033[0m\n' "$line" ;;
+        *)        printf '  \033[1;31m%s\033[0m\n' "$line" ;;
+    esac
+done
+printf '\033[1m=================================================================\033[0m\n'
+printf '  time: %s s\n' "$ELAPSED"
+
+if [ "$PROBLEMS" -eq 0 ]; then
+    printf '\n\033[1;32m  ALL GOOD - ComfyUI is starting on port %s\033[0m\n' "$PORT"
+    printf '\033[1;32m  Connect -> HTTP Service -> :%s\033[0m\n\n' "$PORT"
+else
+    printf '\n\033[1;33m  %s PROBLEM(S) ABOVE - ComfyUI still starts, but some\n' "$PROBLEMS"
+    printf '  nodes in the workflow will be red / models missing.\n'
+    printf '  Full log: %s/provision.log\033[0m\n\n' "$ROOT"
+fi
+
+cd "$COMFY"
+exec python main.py --listen 0.0.0.0 --port "$PORT"