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

installers/nudify/install_flux_ultimate_nudify_v0.19.sh

@@ -0,0 +1,468 @@
+#!/bin/bash
+# ═══════════════════════════════════════════════════════════════════════════════
+# FLUX Ultimate Nudify - Installation Script v0.19
+# ═══════════════════════════════════════════════════════════════════════════════
+# 
+# Core 3-Stage Pipeline:
+#   Stage 1: Clothes Removal (FLUX.1 Kontext Dev)
+#   Stage 2: Anatomy Enhancement (FLUX.1 Dev - Fluxed Up 7.1)
+#   Stage 3: Detail Refinement (FLUX.2 Klein 9B + LoRAs)
+#
+# ═══════════════════════════════════════════════════════════════════════════════
+
+set -e
+
+# Colors
+RED=''
+GREEN=''
+YELLOW=''
+BLUE=''
+MAGENTA=''
+CYAN=''
+NC=''
+
+# Configuration
+COMFY_DIR="${COMFY_DIR:-/workspace/ComfyUI}"
+CIVITAI_TOKEN="${CIVITAI_TOKEN:-eff2ba98a34d6fd7918164753e878bb9}"
+
+# Mirror repo for risky models
+MIRROR_REPO="https://huggingface.co/msrcam/ComfyUI_Models/resolve/main"
+
+# Auto-detect HF token
+if [ -n "$HF_TOKEN" ]; then
+    :
+elif [ -f "$HOME/.cache/huggingface/token" ]; then
+    HF_TOKEN=$(cat "$HOME/.cache/huggingface/token")
+    echo -e "${GREEN}✓ Using cached HuggingFace token${NC}"
+elif [ -f "$HOME/.huggingface/token" ]; then
+    HF_TOKEN=$(cat "$HOME/.huggingface/token")
+    echo -e "${GREEN}✓ Using cached HuggingFace token${NC}"
+else
+    echo -e "${YELLOW}! No HF token found - gated models may fail${NC}"
+    HF_TOKEN=""
+fi
+
+# Banner
+echo -e "${CYAN}═══════════════════════════════════════════════════════════════════════════════${NC}"
+echo -e "${CYAN}  FLUX ULTIMATE NUDIFY v0.19${NC}"
+echo -e "${CYAN}═══════════════════════════════════════════════════════════════════════════════${NC}"
+echo ""
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# INSTALL ARIA2C IF MISSING
+# ═══════════════════════════════════════════════════════════════════════════════
+if ! command -v aria2c &> /dev/null; then
+    echo -e "${YELLOW}Installing aria2c...${NC}"
+    apt-get update -qq && apt-get install -y -qq aria2 > /dev/null 2>&1 || true
+fi
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# DOWNLOAD FUNCTION WITH FALLBACK
+# ═══════════════════════════════════════════════════════════════════════════════
+dl() {
+    local url="$1"
+    local dest="$2"
+    local desc="$3"
+    local fallback_url="${4:-}"
+    
+    if [ -f "$dest" ]; then
+        echo -e "${GREEN}✓ Already exists: ${desc}${NC}"
+        return 0
+    fi
+    
+    mkdir -p "$(dirname "$dest")"
+    echo -e "${YELLOW}↓ ${desc}${NC}"
+    
+    # Add civitai token if needed
+    if [[ "$url" == *"civitai.com"* ]]; then
+        url="${url}?token=${CIVITAI_TOKEN}"
+    fi
+    
+    # Try primary URL
+    if command -v aria2c &> /dev/null; then
+        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}✓ Downloaded: ${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}✓ Downloaded: ${desc}${NC}"
+                return 0
+            }
+        fi
+    fi
+    
+    # Try wget as backup
+    wget -q --show-progress -O "$dest" "$url" 2>/dev/null && {
+        echo -e "${GREEN}✓ Downloaded: ${desc}${NC}"
+        return 0
+    }
+    
+    # Try fallback URL if provided
+    if [ -n "$fallback_url" ]; then
+        echo -e "${YELLOW}  Trying fallback URL...${NC}"
+        aria2c -x 16 -s 16 -k 1M -c --console-log-level=warn \
+            -d "$(dirname "$dest")" -o "$(basename "$dest")" "$fallback_url" 2>/dev/null && {
+            echo -e "${GREEN}✓ Downloaded from fallback: ${desc}${NC}"
+            return 0
+        }
+        wget -q --show-progress -O "$dest" "$fallback_url" 2>/dev/null && {
+            echo -e "${GREEN}✓ Downloaded from fallback: ${desc}${NC}"
+            return 0
+        }
+    fi
+    
+    echo -e "${RED}✗ Failed: ${desc}${NC}"
+    rm -f "$dest" 2>/dev/null
+    return 1
+}
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 1: VERIFY COMFYUI
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "
+${YELLOW}[1/8] Verifying ComfyUI Installation...${NC}"
+
+if [ ! -d "$COMFY_DIR" ]; then
+    echo -e "${RED}Error: ComfyUI not found at $COMFY_DIR${NC}"
+    exit 1
+fi
+
+echo -e "${GREEN}✓ ComfyUI found at $COMFY_DIR${NC}"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 2: CREATE DIRECTORIES
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "
+${YELLOW}[2/8] Creating Directory Structure...${NC}"
+
+dirs=(
+    "models/checkpoints"
+    "models/unet"
+    "models/diffusion_models"
+    "models/vae"
+    "models/clip"
+    "models/clip/t5"
+    "models/text_encoders"
+    "models/loras"
+    "models/upscale_models"
+    "models/ipadapter"
+    "models/clip_vision"
+    "models/controlnet"
+    "models/facerestore_models"
+    "models/facedetection"
+    "models/insightface/models/buffalo_l"
+    "models/ultralytics/bbox"
+    "models/ultralytics/segm"
+    "models/sams"
+    "models/reswapper"
+    "models/hyperswap"
+    "user/default/workflows"
+)
+
+for dir in "${dirs[@]}"; do
+    mkdir -p "$COMFY_DIR/$dir"
+done
+
+echo -e "${GREEN}✓ Directories created${NC}"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 3: INSTALL CUSTOM NODES
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "
+${YELLOW}[3/8] Installing Custom Nodes...${NC}"
+
+cd "$COMFY_DIR/custom_nodes"
+
+# Core nodes
+for repo in \
+    "https://github.com/ltdrdata/ComfyUI-Manager.git" \
+    "https://github.com/rgthree/rgthree-comfy.git" \
+    "https://github.com/cubiq/ComfyUI_essentials.git" \
+    "https://github.com/ltdrdata/ComfyUI-Impact-Pack.git" \
+    "https://github.com/ltdrdata/ComfyUI-Impact-Subpack.git" \
+    "https://github.com/Gourieff/ComfyUI-ReActor.git" \
+    "https://github.com/cubiq/ComfyUI_IPAdapter_plus.git" \
+    "https://github.com/Fannovel16/comfyui_controlnet_aux.git" \
+    "https://github.com/ssitu/ComfyUI_UltimateSDUpscale.git" \
+    "https://github.com/kijai/ComfyUI-segment-anything-2.git" \
+    "https://github.com/kijai/ComfyUI-KJNodes.git" \
+    "https://github.com/spacepxl/ComfyUI-Image-Filters.git" \
+    "https://github.com/EllangoK/ComfyUI-post-processing-nodes.git" \
+    "https://github.com/chrisgoringe/cg-use-everywhere.git" \
+    "https://github.com/pythongosssss/ComfyUI-Custom-Scripts.git"
+do
+    name=$(basename "$repo" .git)
+    if [ ! -d "$name" ]; then
+        echo -e "${GREEN}→ Installing $name${NC}"
+        git clone "$repo" 2>/dev/null || true
+        [ -f "$name/requirements.txt" ] && pip install -r "$name/requirements.txt" --break-system-packages 2>/dev/null || true
+    else
+        echo -e "${GREEN}✓ $name already installed${NC}"
+    fi
+done
+
+# Run Impact Pack install script
+[ -f "ComfyUI-Impact-Pack/install.py" ] && python ComfyUI-Impact-Pack/install.py 2>/dev/null || true
+
+echo -e "${GREEN}✓ All custom nodes installed${NC}"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 4: CORE PIPELINE MODELS
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "
+${YELLOW}[4/8] Downloading Core Pipeline Models...${NC}"
+
+# Stage 1: Kontext
+echo -e "
+${MAGENTA}═══ Stage 1: 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/unet/flux1-kontext-dev.safetensors" \
+   "FLUX.1 Kontext Dev"
+
+dl "https://huggingface.co/msrcam/ComfyUI_Models/resolve/main/loras/kontext/clothes_remover_v0.safetensors" \
+   "$COMFY_DIR/models/loras/clothes_remover_v0.safetensors" \
+   "Clothes Remover LoRA"
+
+# Stage 2: Flux Dev
+echo -e "
+${MAGENTA}═══ Stage 2: FLUX.1 Dev ═══${NC}"
+
+dl "$MIRROR_REPO/checkpoints/flux/fluxed-up-7.1-fp16.safetensors" \
+   "$COMFY_DIR/models/checkpoints/fluxed-up-7.1-fp16.safetensors" \
+   "Fluxed Up 7.1 FP16" \
+   "https://civitai.com/api/download/models/2577735?type=Model&format=SafeTensor&size=pruned&fp=fp16"
+
+dl "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors" \
+   "$COMFY_DIR/models/clip/clip_l.safetensors" \
+   "CLIP-L"
+
+dl "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp8_e4m3fn.safetensors" \
+   "$COMFY_DIR/models/clip/t5/t5xxl_fp8_e4m3fn.safetensors" \
+   "T5-XXL FP8"
+
+# Stage 3: Klein 9B
+echo -e "
+${MAGENTA}═══ Stage 3: 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 FP8"
+
+dl "https://huggingface.co/Comfy-Org/flux2-klein-9B/resolve/main/split_files/text_encoders/qwen_3_8b_fp8mixed.safetensors" \
+   "$COMFY_DIR/models/text_encoders/qwen_3_8b_fp8mixed.safetensors" \
+   "Qwen 3 8B FP8"
+
+dl "https://huggingface.co/Comfy-Org/flux2-dev/resolve/main/split_files/vae/flux2-vae.safetensors" \
+   "$COMFY_DIR/models/vae/flux2-vae.safetensors" \
+   "FLUX2 VAE"
+
+# FLUX.2 Dev main model (gated - requires HF token)
+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"
+
+# FLUX.2 Text Encoder - Mistral 3 Small
+dl "https://huggingface.co/Comfy-Org/flux2-dev/resolve/main/split_files/text_encoders/mistral_3_small_flux2_bf16.safetensors" \
+   "$COMFY_DIR/models/text_encoders/mistral_3_small_flux2_bf16.safetensors" \
+   "Mistral 3 Small (FLUX.2 Text Encoder)"
+
+# Shared VAE
+dl "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/ae.safetensors" \
+   "$COMFY_DIR/models/vae/ae.safetensors" \
+   "FLUX.1 VAE"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 5: LoRAs
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "
+${YELLOW}[5/8] Downloading LoRAs...${NC}"
+
+# Download from mirror repo (consolidated)
+echo -e "${CYAN}═══ Downloading LoRAs from mirror ═══${NC}"
+
+for lora in \
+    "flux1dev/flux1dev_nudity_female_v2.safetensors" \
+    "flux1dev/flux1dev_anatomy_female.safetensors" \
+    "flux1dev/flux1dev_nipple_default.safetensors" \
+    "flux1dev/flux1dev_pussy_default.safetensors" \
+    "flux2klein/flux2klein_nsfw.safetensors" \
+    "flux2klein/flux2klein_nipple.safetensors" \
+    "flux2klein/flux2klein_pussy.safetensors" \
+    "flux2klein/flux2klein_body_version_a.safetensors"
+do
+    name=$(basename "$lora")
+    dl "$MIRROR_REPO/loras/$lora" \
+       "$COMFY_DIR/models/loras/$name" \
+       "$name"
+done
+
+# Detail LoRAs - mirror first
+dl "$MIRROR_REPO/loras/add_detail.safetensors" \
+   "$COMFY_DIR/models/loras/add_detail.safetensors" \
+   "Add Detail LoRA" \
+   "https://civitai.com/api/download/models/135867"
+
+dl "$MIRROR_REPO/loras/skin_detail.safetensors" \
+   "$COMFY_DIR/models/loras/skin_detail.safetensors" \
+   "Skin Detail LoRA" \
+   "https://civitai.com/api/download/models/87153"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 6: FACE & DETECTION MODELS
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "
+${YELLOW}[6/8] Downloading Face & Detection Models...${NC}"
+
+# Face Restore - mirror first
+echo -e "${CYAN}═══ Face Restore ═══${NC}"
+
+dl "$MIRROR_REPO/facerestore/codeformer.pth" \
+   "$COMFY_DIR/models/facerestore_models/codeformer.pth" \
+   "CodeFormer" \
+   "https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth"
+
+dl "$MIRROR_REPO/facerestore/GFPGANv1.4.pth" \
+   "$COMFY_DIR/models/facerestore_models/GFPGANv1.4.pth" \
+   "GFPGAN v1.4" \
+   "https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth"
+
+dl "$MIRROR_REPO/facerestore/GPEN-BFR-1024.onnx" \
+   "$COMFY_DIR/models/facerestore_models/GPEN-BFR-1024.onnx" \
+   "GPEN-BFR-1024"
+
+dl "$MIRROR_REPO/facerestore/GPEN-BFR-2048.onnx" \
+   "$COMFY_DIR/models/facerestore_models/GPEN-BFR-2048.onnx" \
+   "GPEN-BFR-2048"
+
+# Detection models - from mirror
+echo -e "${CYAN}═══ Detection Models ═══${NC}"
+
+dl "$MIRROR_REPO/detection/parsing_parsenet.pth" \
+   "$COMFY_DIR/models/facedetection/parsing_parsenet.pth" \
+   "Parsing ParseNet"
+
+dl "$MIRROR_REPO/detection/detection_Resnet50_Final.pth" \
+   "$COMFY_DIR/models/facedetection/detection_Resnet50_Final.pth" \
+   "Detection Resnet50"
+
+dl "$MIRROR_REPO/detection/face_yolov8m.pt" \
+   "$COMFY_DIR/models/ultralytics/bbox/face_yolov8m.pt" \
+   "YOLOv8m Face"
+
+dl "$MIRROR_REPO/detection/hand_yolov8s.pt" \
+   "$COMFY_DIR/models/ultralytics/bbox/hand_yolov8s.pt" \
+   "YOLOv8s Hand"
+
+dl "$MIRROR_REPO/detection/person_yolov8s-seg.pt" \
+   "$COMFY_DIR/models/ultralytics/segm/person_yolov8s-seg.pt" \
+   "YOLOv8s Person Seg"
+
+dl "$MIRROR_REPO/detection/sam_vit_l_0b3195.pth" \
+   "$COMFY_DIR/models/sams/sam_vit_l_0b3195.pth" \
+   "SAM ViT-L"
+
+# InsightFace - from mirror
+echo -e "${CYAN}═══ InsightFace ═══${NC}"
+
+for model in 1k3d68.onnx 2d106det.onnx det_10g.onnx genderage.onnx w600k_r50.onnx; do
+    dl "$MIRROR_REPO/insightface/buffalo_l/$model" \
+       "$COMFY_DIR/models/insightface/models/buffalo_l/$model" \
+       "Buffalo_L $model"
+done
+
+# Swappers - from mirror
+echo -e "${CYAN}═══ Swappers ═══${NC}"
+
+dl "$MIRROR_REPO/swappers/reswapper_128.onnx" \
+   "$COMFY_DIR/models/reswapper/reswapper_128.onnx" \
+   "ReSwapper 128"
+
+dl "$MIRROR_REPO/swappers/reswapper_256.onnx" \
+   "$COMFY_DIR/models/reswapper/reswapper_256.onnx" \
+   "ReSwapper 256"
+
+dl "$MIRROR_REPO/swappers/hyperswap_1a_256.onnx" \
+   "$COMFY_DIR/models/hyperswap/hyperswap_1a_256.onnx" \
+   "HyperSwap 1A"
+
+dl "$MIRROR_REPO/swappers/hyperswap_1b_256.onnx" \
+   "$COMFY_DIR/models/hyperswap/hyperswap_1b_256.onnx" \
+   "HyperSwap 1B"
+
+dl "$MIRROR_REPO/swappers/hyperswap_1c_256.onnx" \
+   "$COMFY_DIR/models/hyperswap/hyperswap_1c_256.onnx" \
+   "HyperSwap 1C"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 7: UPSCALE & POST-PROCESSING
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "
+${YELLOW}[7/8] Downloading Upscale & Post-Processing Models...${NC}"
+
+echo -e "${CYAN}═══ Upscalers ═══${NC}"
+
+dl "$MIRROR_REPO/upscalers/4x_NMKD-Siax_200k.pth" \
+   "$COMFY_DIR/models/upscale_models/4x_NMKD-Siax_200k.pth" \
+   "4x NMKD-Siax"
+
+dl "$MIRROR_REPO/upscalers/4x-UltraSharp.pth" \
+   "$COMFY_DIR/models/upscale_models/4x-UltraSharp.pth" \
+   "4x UltraSharp"
+
+dl "$MIRROR_REPO/upscalers/RealESRGAN_x4.pth" \
+   "$COMFY_DIR/models/upscale_models/RealESRGAN_x4.pth" \
+   "RealESRGAN x4"
+
+echo -e "${CYAN}═══ IP-Adapter ═══${NC}"
+
+dl "$MIRROR_REPO/ipadapter/ip-adapter-plus_sdxl_vit-h.safetensors" \
+   "$COMFY_DIR/models/ipadapter/ip-adapter-plus_sdxl_vit-h.safetensors" \
+   "IP-Adapter Plus SDXL"
+
+dl "$MIRROR_REPO/ipadapter/ip-adapter-plus-face_sdxl_vit-h.safetensors" \
+   "$COMFY_DIR/models/ipadapter/ip-adapter-plus-face_sdxl_vit-h.safetensors" \
+   "IP-Adapter Plus Face"
+
+echo -e "${CYAN}═══ CLIP Vision ═══${NC}"
+
+dl "$MIRROR_REPO/clip/CLIP-ViT-H-14-laion2B-s32B-b79K.safetensors" \
+   "$COMFY_DIR/models/clip_vision/CLIP-ViT-H-14-laion2B-s32B-b79K.safetensors" \
+   "CLIP ViT-H-14"
+
+echo -e "${CYAN}═══ ControlNet ═══${NC}"
+
+dl "$MIRROR_REPO/controlnet/flux-depth-controlnet-v3.safetensors" \
+   "$COMFY_DIR/models/controlnet/flux-depth-controlnet-v3.safetensors" \
+   "FLUX Depth ControlNet v3"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 8: WORKFLOW
+# ═══════════════════════════════════════════════════════════════════════════════
+# STEP 8: WORKFLOW INSTALLATION
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "\n${YELLOW}[8/8] Installing Workflow...${NC}"
+
+curl -sL "https://huggingface.co/msrcam/ComfyUI_AutoInstallers/resolve/main/installers/nudify/Flux_Ultimate_Nudify_v0.19.json" \
+    -o "$COMFY_DIR/user/default/workflows/Flux_Ultimate_Nudify_v0.19.json"
+
+echo -e "${GREEN}✓ Workflow installed${NC}"
+
+# ═══════════════════════════════════════════════════════════════════════════════
+# DONE
+# ═══════════════════════════════════════════════════════════════════════════════
+echo -e "\n${GREEN}═══════════════════════════════════════════════════════════════════════════════${NC}"
+echo -e "${GREEN}  ✓ FLUX Ultimate Nudify v0.19 - Installation Complete!${NC}"
+echo -e "${GREEN}═══════════════════════════════════════════════════════════════════════════════${NC}"
+
+echo -e "\n${CYAN}To start ComfyUI:${NC}"
+echo "  cd $COMFY_DIR && python main.py --listen 0.0.0.0 --port 8188"
+
+echo -e "\n${CYAN}Workflow location:${NC}"
+echo "  $COMFY_DIR/user/default/workflows/Flux_Ultimate_Nudify_v0.19.json"
+echo ""