Update setup_comfy_wan222.sh

setup_comfy_wan222.sh

@@ -26,6 +26,7 @@ MODELS_DIR="$COMFY_DIR/models"
 COMFY_HOST="${COMFY_HOST:-0.0.0.0}"
 COMFY_PORT="${COMFY_PORT:-8818}"
 CIVITAI_TOKEN="${CIVITAI_TOKEN:-4fcb2834969399006a736ee402b061e5}"
+HF_TOKEN="${HF_TOKEN:-}"  # Token opcional do HuggingFace para login
 
 # Configurações de performance
 export MAX_JOBS=32
@@ -89,7 +90,7 @@ add_civitai_token() {
     fi
 }
 
-# Download HuggingFace usando novo comando hf
+# Download HuggingFace usando huggingface-cli com hf_transfer
 download_hf() {
     local repo="$1"
     local file_path="$2"
@@ -111,53 +112,27 @@ download_hf() {
     
     log_info "Baixando de HF: $filename"
     
-    # Criar diretório temporário para download
-    local temp_dir=$(mktemp -d)
-    
-    # Usar novo comando hf com HF Transfer habilitado
-    if HF_HUB_ENABLE_HF_TRANSFER=1 hf download "$repo" "$file_path" \
-        --local-dir "$temp_dir" \
+    # Usar huggingface-cli com hf_transfer habilitado para velocidade máxima
+    if HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo" "$file_path" \
+        --local-dir "$target_dir" \
         --local-dir-use-symlinks False \
         --resume-download 2>/dev/null; then
         
-        # Encontrar o arquivo baixado (pode estar em subdiretório)
-        local downloaded_file=$(find "$temp_dir" -name "$filename" -type f 2>/dev/null | head -1)
-        
-        if [ -z "$downloaded_file" ]; then
-            # Tentar encontrar qualquer arquivo .safetensors
-            downloaded_file=$(find "$temp_dir" -name "*.safetensors" -type f 2>/dev/null | head -1)
-        fi
-        
-        if [ -n "$downloaded_file" ]; then
-            mv "$downloaded_file" "$target_file"
-            rm -rf "$temp_dir"
+        # Verificar se arquivo foi baixado
+        if [ -f "$target_file" ]; then
             log_success "Download concluído: $filename"
             return 0
         fi
-    fi
-    
-    # Fallback para huggingface-cli se hf falhar
-    if huggingface-cli download "$repo" "$file_path" \
-        --local-dir "$temp_dir" \
-        --local-dir-use-symlinks False \
-        --resume-download 2>/dev/null; then
-        
-        # Procurar arquivo baixado
-        local downloaded_file=$(find "$temp_dir" -name "$filename" -type f 2>/dev/null | head -1)
-        
-        if [ -z "$downloaded_file" ]; then
-            downloaded_file=$(find "$temp_dir" -name "*.safetensors" -type f 2>/dev/null | head -1)
-        fi
         
-        if [ -n "$downloaded_file" ]; then
+        # Procurar arquivo em subdiretórios se não estiver no local esperado
+        local downloaded_file=$(find "$target_dir" -name "$filename" -type f 2>/dev/null | head -1)
+        if [ -n "$downloaded_file" ] && [ "$downloaded_file" != "$target_file" ]; then
             mv "$downloaded_file" "$target_file"
-            rm -rf "$temp_dir"
-            log_success "Download concluído (fallback): $filename"
+            log_success "Download concluído e movido: $filename"
             return 0
         fi
     fi
     
-    rm -rf "$temp_dir"
     log_error "Falha ao baixar: $filename"
     return 1
 }
@@ -336,12 +311,21 @@ log_success "Dependências OK"
 log_info "[2/8] Preparando ferramentas Python..."
 python3 -m pip install --upgrade pip wheel setuptools -q
 
-# Instalar huggingface_hub com hf_transfer
-python3 -m pip install --upgrade "huggingface_hub[hf_transfer]>=0.26.0" comfy-cli -q
+# Instalar huggingface_hub com CLI e hf_transfer para velocidade máxima
+python3 -m pip install --upgrade "huggingface_hub[cli]>=0.26.0" hf_transfer comfy-cli -q
+
+# Configurar token HF se disponível (opcional)
+HF_TOKEN="${HF_TOKEN:-}"
+if [ -n "$HF_TOKEN" ]; then
+    echo "$HF_TOKEN" | huggingface-cli login --token
+    log_success "Login HF configurado"
+fi
 
 # Verificar se hf está disponível
-if ! command_exists hf; then
-    log_warn "Comando 'hf' não encontrado, usando huggingface-cli como fallback"
+if command_exists huggingface-cli; then
+    log_success "huggingface-cli disponível"
+else
+    log_warn "huggingface-cli não encontrado"
 fi
 
 log_success "Ferramentas instaladas"
@@ -405,26 +389,12 @@ log_info "[7/8] 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
-
+# Instalar nodes sequencialmente para evitar travamentos
 for repo_url in "${CUSTOM_NODES[@]}"; do
     node_name=$(basename "$repo_url")
-    
-    # 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
+    clone_or_update "$repo_url" "$CN_DIR/$node_name"
 done
 
-# Aguardar últimos processos
-wait
-
 log_success "Custom nodes instalados"
 
 # [8] Verificação final