Update wanForProduction.sh

wanForProduction.sh

@@ -38,22 +38,6 @@ export PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"
 # Lista de downloads (formato: URL|TIPO|NOME_OPCIONAL)
 DOWNLOAD_FILES=(
     # Wan 2.2 principais (HF)
-    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_HIGH_fp16.safetensors|loras|Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_HIGH_fp16.safetensors"
-    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_LOW_fp16.safetensors|loras|Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_LOW_fp16.safetensors"
-    "hf://Kijai/WanVideo_comfy_fp8_scaled/I2V/Wan2_2-I2V-A14B-HIGH_fp8_e5m2_scaled_KJ.safetensors|diffusion_models|Wan2_2-I2V-A14B-HIGH_fp8_e5m2_scaled_KJ.safetensors"
-    "hf://Kijai/WanVideo_comfy_fp8_scaled/I2V/Wan2_2-I2V-A14B-LOW_fp8_e5m2_scaled_KJ.safetensors|diffusion_models|Wan2_2-I2V-A14B-LOW_fp8_e5m2_scaled_KJ.safetensors"
-    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_I2V-A14B-4steps-lora_HIGH_fp16.safetensors|loras|Wan2.2-Lightning_I2V-A14B-4steps-lora_HIGH_fp16.safetensors"
-    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_I2V-A14B-4steps-lora_LOW_fp16.safetensors|loras|Wan2.2-Lightning_I2V-A14B-4steps-lora_LOW_fp16.safetensors"
-    "hf://Kijai/WanVideo_comfy/Wan2_1_VAE_fp32.safetensors|vae|Wan2_1_VAE_fp32.safetensors"
-    "hf://Kijai/WanVideo_comfy/umt5-xxl-enc-bf16.safetensors|text_encoders|umt5-xxl-enc-bf16.safetensors"
-    "hf://Kijai/WanVideo_comfy/Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank256_bf16.safetensors|loras|lightx2v_I2V_14B_480p_cfg_step_distill_rank256_bf16.safetensors"
-    "hf://Comfy-Org/Wan_2.2_ComfyUI_Repackaged/split_files/vae/wan_2.1_vae.safetensors|vae|wan_2.1_vae.safetensors"
-    "hf://Comfy-Org/Wan_2.2_ComfyUI_Repackaged/split_files/text_encoders/umt5_xxl_fp16.safetensors|text_encoders|umt5_xxl_fp16.safetensors"
-    "hf://Kijai/WanVideo_comfy/open-clip-xlm-roberta-large-vit-huge-14_visual_fp32.safetensors|clip_vision|open-clip-xlm-roberta-large-vit-huge-14_visual_fp32.safetensors"
-    "hf://Comfy-Org/Wan_2.1_ComfyUI_repackaged/split_files/clip_vision/clip_vision_h.safetensors|clip_vision|clip_vision_h.safetensors"
-    "hf://ABDALLALSWAITI/Upscalers/anime/2x-AnimeSharpV2_MoSR_Soft.pth|upscale_models|2x-AnimeSharpV2_MoSR_Soft.pth"
-
-    # Wan 2.2 extras (Mega.nz)
     "mega://https://mega.nz/file/QFxREJpZ#0dge8TyRgb52lMFKdsk-Ac-rwcoG49hvBHvaqNsZkuk|loras|"
     "mega://https://mega.nz/file/lIhQETjT#nKlQ01Euw6fUxo8P5bfwlQrxAkWEedXQQOqHv5u4yrM|loras|"
     "mega://https://mega.nz/file/UIAESACY#Qlei1Pj5Nwno3Sz-wsTekvD-YgCkN1A5QbFKoRHXG9E|loras|"
@@ -67,6 +51,15 @@ DOWNLOAD_FILES=(
     "mega://https://mega.nz/file/hZhhhCQS#4je0Im2g1kEvTCuPotjVWkA-g6XICkqMb-mKxMt6R3Y|loras|"
     "mega://https://mega.nz/file/qdQwAJSC#rPY2Hg4x-_RbHiHUBFW-YU8xZUqPpq5Gpl1uPFiJvW4|upscale_models|"
     "mega://https://mega.nz/file/KIh2marA#CehVNXxQhGfp1bqUL0K1YPWNqrCUoLPCZGq2W6XqmWo|upscale_models|"
+    "https://huggingface.co/Kijai/WanVideo_comfy_fp8_scaled/resolve/main/I2V/Wan2_2-I2V-A14B-LOW_fp8_e4m3fn_scaled_KJ.safetensors|diffusion_models|"
+    "https://huggingface.co/Kijai/WanVideo_comfy_fp8_scaled/resolve/main/I2V/Wan2_2-I2V-A14B-HIGH_fp8_e4m3fn_scaled_KJ.safetensors|diffusion_models|"
+    "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/LoRAs/Wan22-Lightning/Wan22_A14B_T2V_HIGH_Lightning_4steps_lora_250928_rank128_fp16.safetensors|loras|"
+    "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/LoRAs/Wan22-Lightning/Wan22_A14B_T2V_LOW_Lightning_4steps_lora_250928_rank64_fp16.safetensors|loras|"
+    "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/LoRAs/AniSora/Wan2_2_I2V_AniSora_3_2_HIGH_rank_64_fp16.safetensors|loras|"
+    "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/Wan2_1_VAE_bf16.safetensors|vae|"
+    "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/open-clip-xlm-roberta-large-vit-huge-14_visual_fp16.safetensors|clip_vision|"
+    "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/umt5-xxl-enc-bf16.safetensors|text_encoders|"
+    "https://huggingface.co/ABDALLALSWAITI/Upscalers/resolve/main/general/Swin2SR_ClassicalSR_X2_64.pth|upscale_models|"
 )
 
 # Custom nodes (do CODE2)
@@ -116,8 +109,12 @@ download_hf() {
         fi
     fi
 
-    IFS='/' read -r owner repo rest <<< "$hf_path"
-    if [[ -z "$owner" || -z "$repo" || -z "$rest" ]]; then
+    local owner="${hf_path%%/*}"
+    local remainder="${hf_path#*/}"
+    local repo="${remainder%%/*}"
+    local file_path="${remainder#*/}"
+
+    if [[ -z "$owner" || -z "$repo" || -z "$file_path" || "$file_path" == "$remainder" ]]; then
         log_error "Caminho HuggingFace inválido: $hf_path"
         return 1
     fi
@@ -129,13 +126,13 @@ download_hf() {
         success=false
 
         if command -v hf >/dev/null 2>&1; then
-            if HF_HUB_ENABLE_HF_TRANSFER=1 hf download "$repo_id" "$rest" --local-dir "$temp_dir" --quiet; then
+            if HF_HUB_ENABLE_HF_TRANSFER=1 hf download "$repo_id" "$file_path" --local-dir "$temp_dir" --quiet; then
                 success=true
             fi
         fi
 
         if [[ "$success" == false ]]; then
-            local direct_url="https://huggingface.co/${repo_id}/resolve/main/${rest}"
+            local direct_url="https://huggingface.co/${repo_id}/resolve/main/${file_path}"
             if command -v aria2c >/dev/null 2>&1; then
                 local aria_cmd=(aria2c -c -s 16 -x 16 -k 1M --console-log-level=warn --summary-interval=10 --dir "$temp_dir" --out "$filename")
                 if [[ -n "$HF_TOKEN" ]]; then
@@ -174,8 +171,12 @@ download_hf() {
             if [[ -n "$downloaded" ]]; then
                 mv "$downloaded" "$target_dir/$filename"
                 rm -rf "$temp_dir"
-                log_success "Download concluído: $filename"
-                return 0
+                if [[ -s "$target_dir/$filename" ]]; then
+                    log_success "Download concluído: $filename"
+                    return 0
+                fi
+                log_warn "Arquivo vazio após download: $filename"
+                rm -f "$target_dir/$filename"
             fi
         fi
 
@@ -235,8 +236,12 @@ download_file() {
             fi
         fi
         if [[ "$success" == true ]]; then
-            log_success "Download concluído: $final_name"
-            return 0
+            if [[ -s "$target_dir/$final_name" ]]; then
+                log_success "Download concluído: $final_name"
+                return 0
+            fi
+            log_warn "Arquivo vazio após download: $final_name"
+            rm -f "$target_dir/$final_name"
         fi
         log_warn "Tentativa $attempt falhou para $final_name"
         sleep $((2 * attempt))
@@ -430,4 +435,4 @@ log_info "Iniciando ComfyUI em http://${COMFY_HOST}:${COMFY_PORT}"
 log_info "========================================="
 
 cd "$COMFY_DIR"
-exec comfy launch -- ${SAGE_FLAG:+$SAGE_FLAG} --listen "$COMFY_HOST" --port "$COMFY_PORT"
+exec comfy launch -- ${SAGE_FLAG:+$SAGE_FLAG} --listen "$COMFY_HOST" --port "$COMFY_PORT"