Update setup_comfy_wan222.sh

setup_comfy_wan222.sh

@@ -279,19 +279,31 @@ process_downloads() {
 clone_or_update() {
     local url="$1"
     local dest="$2"
+    local node_name=$(basename "$dest")
     
+    # Timeout de 60 segundos para operações git
     if [ -d "$dest/.git" ]; then
-        log_info "Atualizando $(basename "$dest")..."
-        git -C "$dest" pull --ff-only 2>/dev/null || true
+        log_info "Atualizando $node_name..."
+        timeout 60 git -C "$dest" pull --ff-only 2>/dev/null || {
+            log_warn "Timeout ou erro ao atualizar $node_name"
+            return 0  # Não falhar, apenas avisar
+        }
     else
-        log_info "Clonando $(basename "$dest")..."
-        git clone --recursive --depth 1 "$url" "$dest" 2>/dev/null || return 1
+        log_info "Clonando $node_name..."
+        timeout 60 git clone --recursive --depth 1 "$url" "$dest" 2>/dev/null || {
+            log_warn "Falha ao clonar $node_name"
+            return 0  # Não falhar, apenas avisar
+        }
     fi
     
-    # Instalar requirements se existir
+    # Instalar requirements se existir (com timeout)
     if [ -f "$dest/requirements.txt" ]; then
-        python3 -m pip install --no-warn-script-location -q -r "$dest/requirements.txt" 2>/dev/null || true
+        timeout 120 python3 -m pip install --no-warn-script-location -q -r "$dest/requirements.txt" 2>/dev/null || {
+            log_warn "Falha ao instalar requirements para $node_name"
+        }
     fi
+    
+    return 0
 }
 
 # -----------------------------
@@ -349,22 +361,28 @@ else
 fi
 log_success "ComfyUI instalado em: $COMFY_DIR"
 
-# [4] Instalar SageAttention (em background)
+# [4] Instalar SageAttention
 log_info "[4/7] Instalando SageAttention..."
-(
-    if command_exists nvcc; then
-        python3 -m pip install "triton>=3.0.0" nvidia-ml-py -q
-        SAGE_DIR="/tmp/SageAttention"
-        rm -rf "$SAGE_DIR"
-        git clone --depth 1 https://github.com/thu-ml/SageAttention.git "$SAGE_DIR" 2>/dev/null
+if command_exists nvcc; then
+    log_info "Instalando dependências do SageAttention..."
+    python3 -m pip install "triton>=3.0.0" nvidia-ml-py -q 2>/dev/null || true
+    
+    SAGE_DIR="/tmp/SageAttention"
+    rm -rf "$SAGE_DIR"
+    
+    if git clone --depth 1 https://github.com/thu-ml/SageAttention.git "$SAGE_DIR" 2>/dev/null; then
         cd "$SAGE_DIR"
-        python3 setup.py install --quiet 2>/dev/null
+        log_info "Compilando SageAttention (pode demorar alguns minutos)..."
+        python3 setup.py install 2>&1 | grep -E "(running|installing|copying)" || true
+        cd - > /dev/null
+        rm -rf "$SAGE_DIR"
         log_success "SageAttention instalado"
     else
-        log_warn "CUDA não detectado, pulando SageAttention"
+        log_warn "Não foi possível clonar SageAttention, continuando sem ele"
     fi
-) &
-SAGE_PID=$!
+else
+    log_warn "CUDA não detectado, pulando SageAttention"
+fi
 
 # [5] Baixar modelos
 log_info "[5/7] Baixando modelos..."
@@ -389,17 +407,28 @@ log_info "[6/7] Instalando custom nodes..."
 CN_DIR="$COMFY_DIR/custom_nodes"
 mkdir -p "$CN_DIR"
 
+# Instalar nodes em paralelo mas com limite
+MAX_PARALLEL=4
+count=0
+
 for repo_url in "${CUSTOM_NODES[@]}"; do
     node_name=$(basename "$repo_url")
-    clone_or_update "$repo_url" "$CN_DIR/$node_name" &
+    
+    # Executar em background
+    (clone_or_update "$repo_url" "$CN_DIR/$node_name") &
+    
+    # Limitar número de processos paralelos
+    ((count++))
+    if [ $((count % MAX_PARALLEL)) -eq 0 ]; then
+        wait  # Aguardar batch atual terminar
+    fi
 done
+
+# Aguardar últimos processos
 wait
 
 log_success "Custom nodes instalados"
 
-# Aguardar SageAttention
-wait $SAGE_PID 2>/dev/null || true
-
 # [7] Verificação final
 echo ""
 log_info "[7/7] Verificando instalação..."