Upload comfywan_optimized.sh

comfywan_optimized.sh

@@ -283,48 +283,52 @@ install_sageattention() {
 download_hf_models() {
     log_info "Iniciando downloads de modelos do HuggingFace em paralelo..."
     
-    # Array de downloads [repo:arquivo:destino] ou [url_direta:destino]
-    local hf_downloads=(
+    local pids=()
+    
+    # Downloads usando HuggingFace CLI
+    local hf_cli_downloads=(
         "Comfy-Org/Wan_2.1_ComfyUI_repackaged:split_files/diffusion_models/wan2.1_i2v_720p_14B_fp16.safetensors:$WORKSPACE_DIR/models/diffusion_models/"
         "Comfy-Org/Wan_2.1_ComfyUI_repackaged:split_files/clip_vision/clip_vision_h.safetensors:$WORKSPACE_DIR/models/clip_vision/"
         "Comfy-Org/Wan_2.1_ComfyUI_repackaged:split_files/vae/wan_2.1_vae.safetensors:$WORKSPACE_DIR/models/vae/"
+    )
+    
+    # Downloads diretos via URL
+    local direct_downloads=(
         "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/text_encoders/umt5_xxl_fp16.safetensors:$WORKSPACE_DIR/models/text_encoders/"
         "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/Wan21_CausVid_14B_T2V_lora_rank32_v2.safetensors:$WORKSPACE_DIR/models/loras/"
     )
     
-    local pids=()
+    # Processa downloads HF CLI
+    for download in "${hf_cli_downloads[@]}"; do
+        IFS=':' read -r repo file dest <<< "$download"
+        log_info "Iniciando download HF: $repo/$file -> $dest"
+        
+        (
+            cd "$dest"
+            HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo" "$file" --local-dir . --local-dir-use-symlinks False
+            log_info "Concluído: $repo/$file"
+        ) &
+        
+        pids+=($!)
+    done
     
-    for download in "${hf_downloads[@]}"; do
-        IFS=':' read -r source dest <<< "$download"
+    # Processa downloads diretos
+    for download in "${direct_downloads[@]}"; do
+        IFS=':' read -r url dest <<< "$download"
+        log_info "Iniciando download direto: $url -> $dest"
         
-        # Verifica se é URL direta ou repo:arquivo
-        if [[ $source == https://* ]]; then
-            # URL direta
-            log_info "Iniciando download direto: $source -> $dest"
-            
-            (
-                cd "$dest"
-                if command_exists wget; then
-                    wget "$source" --content-disposition
-                elif command_exists curl; then
-                    curl -L -O "$source"
-                else
-                    log_error "Nenhuma ferramenta de download disponível"
-                    exit 1
-                fi
-                log_info "Concluído download direto: $source"
-            ) &
-        else
-            # Repo HuggingFace
-            IFS=':' read -r repo file <<< "$source"
-            log_info "Iniciando download HF: $repo/$file -> $dest"
-            
-            (
-                cd "$dest"
-                HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo" "$file" --local-dir . --local-dir-use-symlinks False
-                log_info "Concluído: $repo/$file"
-            ) &
-        fi
+        (
+            cd "$dest"
+            if command_exists wget; then
+                wget "$url" --content-disposition
+            elif command_exists curl; then
+                curl -L -O "$url"
+            else
+                log_error "Nenhuma ferramenta de download disponível"
+                exit 1
+            fi
+            log_info "Concluído download direto: $url"
+        ) &
         
         pids+=($!)
     done
@@ -420,16 +424,6 @@ launch_comfyui() {
 # ============================================================================
 
 main() {
-
-
-    # Passo 8: OTIMIZAÇÃO - Iniciar SageAttention PRIMEIRO (processo mais longo)
-    log_info "================================================="
-    log_info "Iniciando SageAttention em background (processo mais longo - ~5+ minutos)..."
-    log_info "Enquanto isso, continuaremos com as outras instalações..."
-    log_info "================================================="
-    install_sageattention
-
-    
     log_info "Iniciando instalação dinâmica do ComfyUI..."
     log_info "Diretório de instalação base: $INSTALL_BASE_DIR"
     log_info "================================================="
@@ -455,7 +449,12 @@ main() {
     # Passo 7: Instalar dependências do workflow
     install_workflow_dependencies
     
-
+    # Passo 8: OTIMIZAÇÃO - Iniciar SageAttention PRIMEIRO (processo mais longo)
+    log_info "================================================="
+    log_info "Iniciando SageAttention em background (processo mais longo - ~5+ minutos)..."
+    log_info "Enquanto isso, continuaremos com as outras instalações..."
+    log_info "================================================="
+    install_sageattention
     
     # Passo 9: Instalar custom node WanVideoWrapper
     install_wanvideo_wrapper