Upload provision_full.sh with huggingface_hub

provision_full.sh

@@ -1,7 +1,10 @@
 #!/bin/bash
 set -e
 REPO="https://huggingface.co/LuckyOda/comfyui-full-pack/resolve/main"
-CD="${COMFYUI_DIR:-/workspace/ComfyUI}/models"
+COMFY="${COMFYUI_DIR:-/workspace/ComfyUI}"
+CD="$COMFY/models"
+PIP="${VENV_PIP:-/venv/main/bin/pip}"
+[ ! -f "$PIP" ] && PIP="pip"
 
 dl() {
     local f="$1" d="$2"
@@ -12,8 +15,49 @@ dl() {
     wget -q --show-progress -O "$target" "$REPO/$(echo $f | sed 's/ /%20/g')" || true
 }
 
-echo "=== ComfyUI Full Pack Provisioning ==="
-
+echo "=== LuckyOda Full Pack Provisioning ==="
+
+# ============================================
+# CUSTOM NODES
+# ============================================
+echo ">>> Installing custom nodes"
+cd "$COMFY/custom_nodes"
+repos=(
+    "https://github.com/evanspearman/ComfyMath.git"
+    "https://github.com/LeonQ8/ComfyUI-Dynamic-Lora-Scheduler.git"
+    "https://github.com/Comfy-Org/ComfyUI-Manager.git"
+    "https://github.com/ru4ls/ComfyUI_Nano_Banana.git"
+    "https://github.com/aining2022/ComfyUI_Swwan.git"
+    "https://github.com/replicate/comfyui-replicate.git"
+    "https://github.com/un-seen/comfyui-tensorops.git"
+    "https://github.com/thatboymentor/ofmtechclip.git"
+    "https://github.com/rgthree/rgthree-comfy.git"
+    "https://github.com/kijai/ComfyUI-KJNodes.git"
+    "https://github.com/pythongosssss/ComfyUI-Custom-Scripts.git"
+    "https://github.com/yolain/ComfyUI-Easy-Use.git"
+    "https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git"
+    "https://github.com/kijai/ComfyUI-segment-anything-2.git"
+    "https://github.com/Fannovel16/comfyui_controlnet_aux.git"
+    "https://github.com/ltdrdata/ComfyUI-Impact-Pack.git"
+    "https://github.com/chflame163/ComfyUI_LayerStyle.git"
+    "https://github.com/cubiq/ComfyUI_essentials.git"
+)
+for r in "${repos[@]}"; do
+    dir=$(basename "$r" .git)
+    [ -d "$dir" ] && echo "  [SKIP] $dir" && continue
+    echo "  [CLONE] $dir"
+    git clone "$r" --recursive 2>/dev/null || true
+done
+
+echo ">>> Installing pip dependencies"
+for req in $(find "$COMFY/custom_nodes" -maxdepth 2 -name "requirements.txt"); do
+    $PIP install -q -r "$req" 2>/dev/null || true
+done
+$PIP install -q piexif PyWavelets 2>/dev/null || true
+
+# ============================================
+# MODELS
+# ============================================
 echo ">>> checkpoints"
 dl "SDXLNSFW.safetensors" "$CD/checkpoints"
 dl "v1-5-pruned-emaonly-fp16.safetensors" "$CD/checkpoints"
@@ -22,6 +66,10 @@ echo ">>> clip_vision"
 dl "IronSight_V7.safetensors" "$CD/clip_vision"
 dl "clip_vision_h.safetensors" "$CD/clip_vision"
 
+echo ">>> clip"
+dl "umt5_xxl_fp8_e4m3fn_scaled.safetensors" "$CD/clip"
+dl "clip_vision_h.safetensors" "$CD/clip"
+
 echo ">>> detection"
 dl "vitpose_h_wholebody_data.bin" "$CD/detection"
 dl "vitpose_h_wholebody_model.onnx" "$CD/detection"
@@ -42,7 +90,6 @@ dl "wan2.2_animate_14B_bf16.safetensors" "$CD/diffusion_models"
 dl "Wan2_2-Animate-14B_fp8_scaled_e4m3fn_KJ_v2.safetensors" "$CD/diffusion_models"
 dl "z_image_turbo_bf16.safetensors" "$CD/diffusion_models"
 dl "zimage.safetensors" "$CD/diffusion_models"
-dl "z_image_turbo_bf16.safetensors" "$CD/diffusion_models"
 dl "diffusion_pytorch_model-00001-of-00002.safetensors" "$CD/diffusion_models"
 
 echo ">>> loras"
@@ -63,6 +110,9 @@ dl "makima_Z.safetensors" "$CD/loras"
 dl "primary_net_v2.safetensors" "$CD/loras"
 dl "wan2.2_i2v_lightx2v_4steps_lora_v1_high_noise.safetensors" "$CD/loras"
 dl "x_gen_weights.safetensors" "$CD/loras"
+dl "FameGrid_Revolution_ZIB_BOLD_ (1).safetensors" "$CD/loras"
+dl "f6242fb5-cf69-467f-9947-3eb0601295c2-u1_b4f6c962-d35c-4e2a-8e2b-a9fed14b0921.safetensors" "$CD/loras"
+dl "4c79a09e-7ace-4cd7-946d-f638ccf67523-u2_d27d5558-b0b8-4bcc-80eb-8c968a17b751.safetensors" "$CD/loras"
 
 echo ">>> sam2"
 dl "sam2.1_hiera_base_plus-fp16.safetensors" "$CD/sam2"
@@ -84,14 +134,11 @@ dl "qwen_3_8b.safetensors" "$CD/text_encoders"
 dl "umt5.safetensors" "$CD/text_encoders"
 dl "umt5_xxl_fp8_e4m3fn_scaled.safetensors" "$CD/text_encoders"
 
-echo ">>> clip (animator)"
-dl "umt5_xxl_fp8_e4m3fn_scaled.safetensors" "$CD/clip"
-dl "clip_vision_h.safetensors" "$CD/clip"
-
 echo ">>> ultralytics"
 dl "lips_v1.pt" "$CD/ultralytics"
 dl "nipple.pt" "$CD/ultralytics"
 dl "pussyV2.pt" "$CD/ultralytics"
+mkdir -p "$CD/ultralytics/bbox" "$CD/ultralytics/segm"
 dl "face_yolov8m.pt" "$CD/ultralytics/bbox"
 dl "hand_yolov8s.pt" "$CD/ultralytics/bbox"
 dl "person_yolov8m-seg.pt" "$CD/ultralytics/segm"
@@ -113,16 +160,7 @@ dl "flux2-vae.safetensors" "$CD/vae"
 dl "variational_encoder_primary.safetensors" "$CD/vae"
 dl "wan_2.1_vae.safetensors" "$CD/vae"
 
-echo ">>> custom nodes"
-wget -q --show-progress -O /tmp/nodes.tar.gz "$REPO/all_custom_nodes.tar.gz"
-cd "${COMFYUI_DIR:-/workspace/ComfyUI}/custom_nodes"
-tar -xzf /tmp/nodes.tar.gz
-rm -f /tmp/nodes.tar.gz
-
-echo ">>> animator custom nodes"
-wget -q --show-progress -O /tmp/anodes.tar.gz "$REPO/animator_custom_nodes.tar.gz"
-cd "${COMFYUI_DIR:-/workspace/ComfyUI}/custom_nodes"
-tar -xzf /tmp/anodes.tar.gz
-rm -f /tmp/anodes.tar.gz
+echo ">>> Restarting ComfyUI"
+supervisorctl restart comfyui 2>/dev/null || true
 
 echo "=== Full Pack Complete ==="