Update script_nightly.sh

script_nightly.sh

@@ -39,6 +39,14 @@ CHECKPOINT_MODELS=(
 CLIP_VISION_MODELS=(
     "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/clip_vision/CLIP-ViT-H-14-laion2B-s32B-b79K.safetensors"
 )
+DINO_MODELS=(
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/grounding-dino/groundingdino_swinb_cogcoor.pth"
+)
+FASHION_MODELS=(
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/RMBG/segformer_fashion/config.json"
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/RMBG/segformer_fashion/model.safetensors"
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/RMBG/segformer_fashion/preprocessor_config.json"
+)
 IPADAPTER_MODELS=(
     "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/ipadapter/ip-adapter-plus-face_sdxl_vit-h.safetensors"
     "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/ipadapter/ip-adapter-plus_sdxl_vit-h.safetensors"
@@ -47,6 +55,15 @@ LORAS_MODELS=(
     "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/loras/Better_detailed_pussy_and_anus_v3.0.safetensors"
     "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/loras/deal360acv_illustrious_006.safetensors"
 )
+RMBG_MODELS=(
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/RMBG/RMBG-2.0/BiRefNet_config.py"
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/RMBG/RMBG-2.0/birefnet.py"
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/RMBG/RMBG-2.0/config.json"
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/RMBG/RMBG-2.0/model.safetensors"
+)
+SAM2_MODELS=(
+    "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/sam2/sam2.1_hiera_large.safetensors"
+)
 SAMS_MODELS=(
     "https://huggingface.co/JCscrew/RisuAI_asset_generator/resolve/main/models/sams/sam_vit_h_4b8939.pth"
 )
@@ -304,9 +321,13 @@ provisioning_step3_downloads() {
     log_step "3" "下載模型與工作流文件"
     download_to_directory "${COMFYUI_DIR}/models/checkpoints" "${CHECKPOINT_MODELS[@]}"
     download_to_directory "${COMFYUI_DIR}/models/clip_vision" "${CLIP_VISION_MODELS[@]}"
+    download_to_directory "${COMFYUI_DIR}/models/grounding-dino" "${DINO_MODELS[@]}"
     download_to_directory "${COMFYUI_DIR}/models/ipadapter" "${IPADAPTER_MODELS[@]}"
     download_to_directory "${COMFYUI_DIR}/models/loras" "${LORAS_MODELS[@]}"
-    download_to_directory "${COMFYUI_DIR}/sams/loras" "${SAMS_MODELS[@]}"
+    download_to_directory "${COMFYUI_DIR}/models/RMBG/RMBG-2.0" "${RMBG_MODELS[@]}"  
+    download_to_directory "${COMFYUI_DIR}/models/RMBG/segformer_fashion" "${FASHION_MODELS[@]}"  
+    download_to_directory "${COMFYUI_DIR}/models/sam2" "${SAM2_MODELS[@]}"
+    download_to_directory "${COMFYUI_DIR}/models/sams" "${SAMS_MODELS[@]}"   
     download_to_directory "${COMFYUI_DIR}/models/ultralytics/bbox" "${BBOX_MODELS[@]}"    
     download_to_directory "${COMFYUI_DIR}/models/upscale_models" "${UPSCALE_MODELS[@]}"
     if [ ${#WORKFLOWS[@]} -gt 0 ]; then
@@ -339,8 +360,12 @@ wait $NODE_PID
 log_step "4" "驗證下載完整性"
 verify_and_retry_downloads "${COMFYUI_DIR}/models/checkpoints" "${CHECKPOINT_MODELS[@]}"
 verify_and_retry_downloads "${COMFYUI_DIR}/models/clip_vision" "${CLIP_VISION_MODELS[@]}"
+verify_and_retry_downloads "${COMFYUI_DIR}/models/grounding-dino" "${DINO_MODELS[@]}"
 verify_and_retry_downloads "${COMFYUI_DIR}/models/ipadapter" "${IPADAPTER_MODELS[@]}"
 verify_and_retry_downloads "${COMFYUI_DIR}/models/loras" "${LORAS_MODELS[@]}"
+verify_and_retry_downloads "${COMFYUI_DIR}/models/RMBG/RMBG-2.0" "${RMBG_MODELS[@]}"  
+verify_and_retry_downloads "${COMFYUI_DIR}/models/RMBG/segformer_fashion" "${FASHION_MODELS[@]}"  
+verify_and_retry_downloads "${COMFYUI_DIR}/models/sam2" "${SAM2_MODELS[@]}"
 verify_and_retry_downloads "${COMFYUI_DIR}/models/sams" "${SAMS_MODELS[@]}"
 verify_and_retry_downloads "${COMFYUI_DIR}/models/ultralytics/bbox" "${BBOX_MODELS[@]}"
 verify_and_retry_downloads "${COMFYUI_DIR}/models/upscale_models" "${UPSCALE_MODELS[@]}"