minimax worker: minimax_start.sh

worker/minimax_start.sh

@@ -0,0 +1,62 @@
+#!/bin/bash
+# Boot script for the MiniMax H3 serverless endpoints (fl2va / ref2va).
+# Runs inside runpod/worker-comfyui:*-base before the stock /start.sh:
+#   1. upgrades ComfyUI to a version that supports MiniMax H3 (int8 convrot),
+#   2. wires model files into ComfyUI's model folders — from RunPod's cached
+#      model mount when present, otherwise downloads from HF to container disk,
+#   3. installs the lora-from-url custom node.
+set -eo pipefail
+
+VARIANT="${MODEL_VARIANT:-fl2va}"
+REPO="nikdevs/minimax-h3-${VARIANT}"
+TARGET="${COMFY_VERSION:-v0.31.0}"
+COMFY=/comfyui
+PY="$(command -v python3 || command -v python)"
+
+echo "[minimax] variant=${VARIANT} comfy_target=${TARGET}"
+
+# --- 1. ComfyUI version ---------------------------------------------------
+cd "$COMFY"
+CUR="$(git describe --tags --exact-match 2>/dev/null || echo unknown)"
+if [ "$CUR" != "$TARGET" ]; then
+  echo "[minimax] upgrading ComfyUI ${CUR} -> ${TARGET}"
+  git fetch --depth 1 origin tag "$TARGET" 2>/dev/null || git fetch --tags origin
+  git checkout -q "$TARGET"
+  "$PY" -m pip install -q -r requirements.txt
+else
+  echo "[minimax] ComfyUI already at ${TARGET}"
+fi
+
+# --- 2. models ------------------------------------------------------------
+mkdir -p "$COMFY/models/diffusion_models" "$COMFY/models/text_encoders" \
+         "$COMFY/models/vae" "$COMFY/models/loras"
+
+CACHE_BASE="/runpod-volume/huggingface-cache/hub/models--nikdevs--minimax-h3-${VARIANT}/snapshots"
+SNAP="$(ls -d "$CACHE_BASE"/*/ 2>/dev/null | head -1 || true)"
+if [ -z "$SNAP" ]; then
+  echo "[minimax] no runpod model cache — downloading ${REPO} (~54GB) via hf_transfer"
+  "$PY" -m pip install -q -U "huggingface_hub[hf_transfer]"
+  SNAP="$(HF_HUB_ENABLE_HF_TRANSFER=1 "$PY" - <<PYEOF
+import os
+from huggingface_hub import snapshot_download
+print(snapshot_download("${REPO}", token=os.environ.get("HF_TOKEN")))
+PYEOF
+)"
+fi
+echo "[minimax] model source: ${SNAP}"
+for sub in diffusion_models text_encoders vae; do
+  if [ -d "${SNAP}/${sub}" ]; then
+    ln -sf "${SNAP}/${sub}"/*.safetensors "$COMFY/models/${sub}/"
+  fi
+done
+ls -l "$COMFY/models/diffusion_models" "$COMFY/models/text_encoders" "$COMFY/models/vae"
+
+# --- 3. lora-from-url node ------------------------------------------------
+"$PY" - <<'PYEOF'
+import urllib.request
+url = "https://huggingface.co/nikdevs/qwen/resolve/main/worker/lora_from_url.py"
+open("/comfyui/custom_nodes/lora_from_url.py", "wb").write(urllib.request.urlopen(url).read())
+print("[minimax] lora_from_url node installed")
+PYEOF
+
+exec /start.sh