Upload installers/nudify/install_flux_ultimate_nudify_v0.22.sh with huggingface_hub

installers/nudify/install_flux_ultimate_nudify_v0.22.sh

@@ -2,9 +2,14 @@
 # ═══════════════════════════════════════════════════════════════════════════════
 # FLUX Ultimate Nudify - Installation Script v0.22
 # ═══════════════════════════════════════════════════════════════════════════════
+# 
+# Uses msrcam mirror repo for reliable downloads
+#
+# ═══════════════════════════════════════════════════════════════════════════════
 
 set -e
 
+# Colors
 RED='\033[0;31m'
 GREEN='\033[0;32m'
 YELLOW='\033[1;33m'
@@ -12,73 +17,67 @@ MAGENTA='\033[0;35m'
 CYAN='\033[0;36m'
 NC='\033[0m'
 
+# Configuration
 COMFY_DIR="${COMFY_DIR:-/workspace/ComfyUI}"
 CIVITAI_TOKEN="${CIVITAI_TOKEN:-eff2ba98a34d6fd7918164753e878bb9}"
+HF_TOKEN="${HF_TOKEN:-}"
 
-# Auto-detect HF token
-if [ -z "$HF_TOKEN" ]; then
-    if [ -f "$HOME/.cache/huggingface/token" ]; then
-        HF_TOKEN=$(cat "$HOME/.cache/huggingface/token")
-        echo -e "${GREEN}✓ Using cached HF token${NC}"
-    elif [ -f "$HOME/.huggingface/token" ]; then
-        HF_TOKEN=$(cat "$HOME/.huggingface/token")
-        echo -e "${GREEN}✓ Using cached HF token${NC}"
-    fi
-fi
-
+# Mirror repo
 MIRROR="https://huggingface.co/msrcam/ComfyUI_Models/resolve/main"
 
+# Banner
 echo -e "${CYAN}═══════════════════════════════════════════════════════════════════════════════${NC}"
 echo -e "${CYAN}  FLUX ULTIMATE NUDIFY v0.22${NC}"
 echo -e "${CYAN}═══════════════════════════════════════════════════════════════════════════════${NC}"
+echo ""
 
+# Install aria2c
 apt-get update -qq && apt-get install -y -qq aria2 > /dev/null 2>&1 || true
 echo -e "${GREEN}✓ aria2c ready${NC}"
 
-# Download function with proper token handling
+# Download function
 dl() {
     local url="$1"
     local dest="$2"
     local desc="$3"
     
-    [ -f "$dest" ] && { echo -e "${GREEN}✓ Exists: ${desc}${NC}"; return 0; }
+    if [ -f "$dest" ]; then
+        echo -e "${GREEN}✓ Exists: ${desc}${NC}"
+        return 0
+    fi
     
     mkdir -p "$(dirname "$dest")"
     echo -e "${YELLOW}↓ ${desc}${NC}"
     
+    local headers=""
     [[ "$url" == *"civitai.com"* ]] && url="${url}?token=${CIVITAI_TOKEN}"
+    [[ "$url" == *"huggingface.co"* ]] && [ -n "$HF_TOKEN" ] && headers="--header=Authorization: Bearer ${HF_TOKEN}"
     
-    # Use token for HuggingFace
-    if [[ "$url" == *"huggingface.co"* ]] && [ -n "$HF_TOKEN" ]; then
-        aria2c -x 16 -s 16 -k 1M -c --console-log-level=warn \
-            --header="Authorization: Bearer ${HF_TOKEN}" \
-            -d "$(dirname "$dest")" -o "$(basename "$dest")" "$url" 2>/dev/null && {
-            echo -e "${GREEN}✓ ${desc}${NC}"
-            return 0
-        }
-    else
-        aria2c -x 16 -s 16 -k 1M -c --console-log-level=warn \
-            -d "$(dirname "$dest")" -o "$(basename "$dest")" "$url" 2>/dev/null && {
-            echo -e "${GREEN}✓ ${desc}${NC}"
-            return 0
-        }
-    fi
+    aria2c -x 16 -s 16 -k 1M -c --console-log-level=warn $headers \
+        -d "$(dirname "$dest")" -o "$(basename "$dest")" "$url" 2>/dev/null && {
+        echo -e "${GREEN}✓ ${desc}${NC}"
+        return 0
+    }
     
     echo -e "${RED}✗ Failed: ${desc}${NC}"
     return 1
 }
 
+# ═══════════════════════════════════════════════════════════════════════════════
+# VERIFY & SETUP
 # ═══════════════════════════════════════════════════════════════════════════════
 echo -e "\n${YELLOW}[1/7] Setup...${NC}"
-[ ! -d "$COMFY_DIR" ] && { echo -e "${RED}ComfyUI not found${NC}"; exit 1; }
+[ ! -d "$COMFY_DIR" ] && { echo -e "${RED}ComfyUI not found at $COMFY_DIR${NC}"; exit 1; }
 echo -e "${GREEN}✓ ComfyUI found${NC}"
 
 mkdir -p "$COMFY_DIR/models"/{checkpoints,unet,diffusion_models,vae,clip,text_encoders,loras}
 mkdir -p "$COMFY_DIR/models"/{upscale_models,insightface/models/buffalo_l,facerestore_models}
-mkdir -p "$COMFY_DIR/models"/{ultralytics/bbox,ultralytics/segm,sams}
+mkdir -p "$COMFY_DIR/models"/{ultralytics/bbox,ultralytics/segm,sams,reactor}
 mkdir -p "$COMFY_DIR/user/default/workflows"
 echo -e "${GREEN}�� Directories created${NC}"
 
+# ═══════════════════════════════════════════════════════════════════════════════
+# CUSTOM NODES
 # ═══════════════════════════════════════════════════════════════════════════════
 echo -e "\n${YELLOW}[2/7] Custom Nodes...${NC}"
 cd "$COMFY_DIR/custom_nodes"
@@ -102,20 +101,31 @@ for dir in ComfyUI-Impact-Pack ComfyUI-ReActor; do
     [ -f "$dir/requirements.txt" ] && pip install -r "$dir/requirements.txt" --break-system-packages -q 2>/dev/null || true
 done
 
+# Remove broken GroundingDino if exists
+rm -rf ComfyUI-GroundingDino 2>/dev/null || true
+
+# Install onnxruntime-gpu for ReActor
+pip install onnxruntime-gpu --break-system-packages -q 2>/dev/null || true
+
+# Fix numpy compatibility
+pip install "numpy<2.4" --break-system-packages -q 2>/dev/null || true
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# FLUX MODELS (from mirror)
 # ═══════════════════════════════════════════════════════════════════════════════
 echo -e "\n${YELLOW}[3/7] FLUX Models...${NC}"
 
-echo -e "\n${MAGENTA}═══ Checkpoints (from mirror) ═══${NC}"
+echo -e "\n${MAGENTA}═══ Checkpoints ═══${NC}"
 dl "$MIRROR/checkpoints/flux/fluxed-up-7.1-fp16.safetensors" \
    "$COMFY_DIR/models/checkpoints/fluxed-up-7.1-fp16.safetensors" \
    "Fluxed Up 7.1"
 
-echo -e "\n${MAGENTA}═══ FLUX.1 Kontext Dev (gated) ═══${NC}"
+echo -e "\n${MAGENTA}═══ FLUX.1 Kontext Dev ═══${NC}"
 dl "https://huggingface.co/black-forest-labs/FLUX.1-Kontext-dev/resolve/main/flux1-kontext-dev.safetensors" \
    "$COMFY_DIR/models/diffusion_models/flux1-kontext-dev.safetensors" \
    "FLUX.1 Kontext Dev"
 
-echo -e "\n${MAGENTA}═══ FLUX.2 Klein 9B (gated) ═══${NC}"
+echo -e "\n${MAGENTA}═══ FLUX.2 Klein 9B ═══${NC}"
 dl "https://huggingface.co/black-forest-labs/FLUX.2-klein-base-9b-fp8/resolve/main/flux-2-klein-base-9b-fp8.safetensors" \
    "$COMFY_DIR/models/unet/flux-2-klein-base-9b-fp8.safetensors" \
    "FLUX.2 Klein 9B"
@@ -124,7 +134,7 @@ dl "https://huggingface.co/Comfy-Org/flux2-klein-9B/resolve/main/split_files/tex
    "$COMFY_DIR/models/text_encoders/qwen_3_8b_fp8mixed.safetensors" \
    "Qwen 3 8B"
 
-echo -e "\n${MAGENTA}═══ FLUX.2 Dev (gated) ═══${NC}"
+echo -e "\n${MAGENTA}═══ FLUX.2 Dev ═══${NC}"
 dl "https://huggingface.co/black-forest-labs/FLUX.2-dev/resolve/main/flux2-dev.safetensors" \
    "$COMFY_DIR/models/diffusion_models/flux2-dev.safetensors" \
    "FLUX.2 Dev"
@@ -151,49 +161,106 @@ dl "https://huggingface.co/Comfy-Org/flux2-dev/resolve/main/split_files/vae/flux
    "FLUX.2 VAE"
 
 # ═══════════════════════════════════════════════════════════════════════════════
-echo -e "\n${YELLOW}[4/7] LoRAs (from mirror)...${NC}"
+# LORAS (from mirror)
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "\n${YELLOW}[4/7] LoRAs...${NC}"
 
+echo -e "${MAGENTA}═══ Flux.1 Dev LoRAs ═══${NC}"
 for lora in flux1dev_nudity_female_v2 flux1dev_anatomy_female flux1dev_nipple_default flux1dev_pussy_default; do
-    dl "$MIRROR/loras/flux1dev/${lora}.safetensors" "$COMFY_DIR/models/loras/${lora}.safetensors" "$lora"
+    dl "$MIRROR/loras/flux1dev/${lora}.safetensors" \
+       "$COMFY_DIR/models/loras/${lora}.safetensors" \
+       "$lora"
 done
 
+echo -e "\n${MAGENTA}═══ Flux.2 Klein LoRAs ═══${NC}"
 for lora in flux2klein_nsfw flux2klein_nipple flux2klein_pussy flux2klein_body_version_a flux2klein_style_2000s; do
-    dl "$MIRROR/loras/flux2klein/${lora}.safetensors" "$COMFY_DIR/models/loras/${lora}.safetensors" "$lora"
+    dl "$MIRROR/loras/flux2klein/${lora}.safetensors" \
+       "$COMFY_DIR/models/loras/${lora}.safetensors" \
+       "$lora"
 done
 
-dl "$MIRROR/loras/kontext/clothes_remover_v0.safetensors" "$COMFY_DIR/models/loras/clothes_remover_v0.safetensors" "Clothes Remover"
+echo -e "\n${MAGENTA}═══ Kontext LoRAs ═══${NC}"
+dl "$MIRROR/loras/kontext/clothes_remover_v0.safetensors" \
+   "$COMFY_DIR/models/loras/clothes_remover_v0.safetensors" \
+   "Clothes Remover v0"
 
 # ═══════��═══════════════════════════════════════════════════════════════════════
-echo -e "\n${YELLOW}[5/7] ReActor & Detectors (from mirror)...${NC}"
+# REACTOR & DETECTORS (from mirror)
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "\n${YELLOW}[5/7] ReActor & Detectors...${NC}"
 
-dl "$MIRROR/swappers/reswapper_128.onnx" "$COMFY_DIR/models/insightface/inswapper_128.onnx" "Inswapper"
-dl "$MIRROR/facerestore/GFPGANv1.4.pth" "$COMFY_DIR/models/facerestore_models/GFPGANv1.4.pth" "GFPGAN"
-dl "$MIRROR/facerestore/codeformer.pth" "$COMFY_DIR/models/facerestore_models/codeformer.pth" "CodeFormer"
+echo -e "${MAGENTA}═══ Face Swap ═══${NC}"
+dl "$MIRROR/swappers/reswapper_128.onnx" \
+   "$COMFY_DIR/models/insightface/inswapper_128.onnx" \
+   "Inswapper 128"
 
+dl "$MIRROR/facerestore/GFPGANv1.4.pth" \
+   "$COMFY_DIR/models/facerestore_models/GFPGANv1.4.pth" \
+   "GFPGAN v1.4"
+
+dl "$MIRROR/facerestore/codeformer.pth" \
+   "$COMFY_DIR/models/facerestore_models/codeformer.pth" \
+   "CodeFormer"
+
+echo -e "\n${MAGENTA}═══ InsightFace buffalo_l ═══${NC}"
 for f in 1k3d68.onnx 2d106det.onnx det_10g.onnx genderage.onnx w600k_r50.onnx; do
-    dl "$MIRROR/insightface/buffalo_l/$f" "$COMFY_DIR/models/insightface/models/buffalo_l/$f" "$f"
+    dl "$MIRROR/insightface/buffalo_l/$f" \
+       "$COMFY_DIR/models/insightface/models/buffalo_l/$f" \
+       "buffalo_l/$f"
 done
 
-dl "$MIRROR/detection/face_yolov8m.pt" "$COMFY_DIR/models/ultralytics/bbox/face_yolov8m.pt" "Face YOLO"
-dl "$MIRROR/detection/hand_yolov8s.pt" "$COMFY_DIR/models/ultralytics/bbox/hand_yolov8s.pt" "Hand YOLO"
-dl "$MIRROR/detection/person_yolov8s-seg.pt" "$COMFY_DIR/models/ultralytics/segm/person_yolov8s-seg.pt" "Person YOLO"
-dl "$MIRROR/detection/sam_vit_l_0b3195.pth" "$COMFY_DIR/models/sams/sam_vit_l_0b3195.pth" "SAM"
-dl "$MIRROR/upscalers/4x_NMKD-Siax_200k.pth" "$COMFY_DIR/models/upscale_models/4x_NMKD-Siax_200k.pth" "4x Upscaler"
+echo -e "\n${MAGENTA}═══ Detectors ═══${NC}"
+dl "$MIRROR/detection/face_yolov8m.pt" \
+   "$COMFY_DIR/models/ultralytics/bbox/face_yolov8m.pt" \
+   "Face YOLOv8m"
+
+dl "$MIRROR/detection/hand_yolov8s.pt" \
+   "$COMFY_DIR/models/ultralytics/bbox/hand_yolov8s.pt" \
+   "Hand YOLOv8s"
+
+dl "$MIRROR/detection/person_yolov8s-seg.pt" \
+   "$COMFY_DIR/models/ultralytics/segm/person_yolov8s-seg.pt" \
+   "Person YOLOv8s-seg"
 
+dl "$MIRROR/detection/sam_vit_l_0b3195.pth" \
+   "$COMFY_DIR/models/sams/sam_vit_l_0b3195.pth" \
+   "SAM ViT-L"
+
+echo -e "\n${MAGENTA}═══ Upscaler ═══${NC}"
+dl "$MIRROR/upscalers/4x_NMKD-Siax_200k.pth" \
+   "$COMFY_DIR/models/upscale_models/4x_NMKD-Siax_200k.pth" \
+   "4x NMKD-Siax"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# WORKFLOW
 # ═══════════════════════════════════════════════════════════════════════════════
 echo -e "\n${YELLOW}[6/7] Workflow...${NC}"
 dl "https://huggingface.co/msrcam/ComfyUI_AutoInstallers/resolve/main/installers/nudify/Flux_Ultimate_Nudify_v1.14.json" \
    "$COMFY_DIR/user/default/workflows/Flux_Ultimate_Nudify.json" \
-   "Workflow"
+   "Workflow v1.14"
 
 # ═══════════════════════════════════════════════════════════════════════════════
-echo -e "\n${YELLOW}[7/7] Symlinks...${NC}"
+# SYMLINKS for Impact Pack paths
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "\n${YELLOW}[7/7] Creating symlinks...${NC}"
+cd "$COMFY_DIR/models"
+
+# SAM
+[ ! -L "sams/sam_vit_l_0b3195.pth" ] && [ -f "sams/sam_vit_l_0b3195.pth" ] || true
+
+# Ultralytics expects specific paths
 ln -sf "$COMFY_DIR/models/ultralytics/bbox/face_yolov8m.pt" "$COMFY_DIR/models/ultralytics/face_yolov8m.pt" 2>/dev/null || true
 ln -sf "$COMFY_DIR/models/ultralytics/bbox/hand_yolov8s.pt" "$COMFY_DIR/models/ultralytics/hand_yolov8s.pt" 2>/dev/null || true
-echo -e "${GREEN}✓ Done${NC}"
 
+echo -e "${GREEN}✓ Symlinks created${NC}"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# DONE
 # ═══════════════════════════════════════════════════════════════════════════════
 echo -e "\n${GREEN}═══════════════════════════════════════════════════════════════════════════════${NC}"
 echo -e "${GREEN}  ✓ FLUX Ultimate Nudify v0.22 - Complete!${NC}"
 echo -e "${GREEN}═══════════════════════════════════════════════════════════════════════════════${NC}"
-echo -e "\n${CYAN}Start:${NC} cd $COMFY_DIR && python main.py --listen 0.0.0.0 --port 8188"
+
+echo -e "\n${CYAN}Start ComfyUI:${NC}"
+echo "  cd $COMFY_DIR && python main.py --listen 0.0.0.0 --port 8188"
+echo ""