Update WANcomfyui_installer_fixed.sh

WANcomfyui_installer_fixed.sh

@@ -1,9 +1,8 @@
 #!/bin/bash
 
 # ============================================================================
-# Instalador Dinâmico para ComfyUI com Wan 2.2
-# Instala o ComfyUI e suas dependências no diretório atual.
-# Versão: 2.0 - Corrigida e Otimizada
+# Instalador Otimizado para ComfyUI com Wan 2.2
+# Versão 3.0 - Corrigida com instalação manual de custom nodes
 # ============================================================================
 
 set -e
@@ -13,18 +12,16 @@ set -o pipefail
 # CONFIGURAÇÕES GLOBAIS
 # ============================================================================
 
-# URL do workflow a ser baixado
-readonly WORKFLOW_URL="https://huggingface.co/adbrasi/akarris_trainer/resolve/main/Wan_GenerationBase.json"
-
-# --- Caminhos Dinâmicos ---
+# Diretórios base
 readonly INSTALL_BASE_DIR="$(pwd)"
 readonly WORKSPACE_DIR="$INSTALL_BASE_DIR/ComfyUI"
 readonly WORKFLOW_FILENAME="workflow.json"
 readonly WORKFLOW_FILE="$INSTALL_BASE_DIR/$WORKFLOW_FILENAME"
-
-# Arquivo de log
 readonly LOG_FILE="$INSTALL_BASE_DIR/install.log"
 
+# URL do workflow
+readonly WORKFLOW_URL="https://huggingface.co/adbrasi/akarris_trainer/resolve/main/Wan_GenerationBase.json"
+
 # ============================================================================
 # FUNÇÕES DE UTILIDADE
 # ============================================================================
@@ -75,7 +72,7 @@ check_requirements() {
     
     if [ ${#missing_deps[@]} -gt 0 ]; then
         log_error "Dependências faltando: ${missing_deps[*]}"
-        log_error "Por favor, instale as dependências faltando e execute o script novamente."
+        log_error "Por favor, instale as dependências e execute o script novamente."
         exit 1
     fi
     
@@ -92,7 +89,7 @@ check_requirements() {
 }
 
 # ============================================================================
-# FUNÇÃO DE LIMPEZA
+# LIMPEZA DE INSTALAÇÃO EXISTENTE
 # ============================================================================
 
 cleanup_existing_installation() {
@@ -111,7 +108,7 @@ cleanup_existing_installation() {
             exit 1
         fi
     else
-        log_info "Nenhuma instalação anterior encontrada no local."
+        log_info "Nenhuma instalação anterior encontrada."
     fi
 }
 
@@ -131,12 +128,11 @@ install_comfy_cli() {
     # Adiciona ~/.local/bin ao PATH se necessário
     if [ -d "$HOME/.local/bin" ] && [[ ":$PATH:" != *":$HOME/.local/bin:"* ]]; then
         export PATH="$HOME/.local/bin:$PATH"
-        log_info "Adicionado $HOME/.local/bin ao PATH para esta sessão."
+        log_info "Adicionado $HOME/.local/bin ao PATH."
     fi
     
     if ! command_exists comfy; then
-        log_error "Comando 'comfy' não encontrado no PATH após a instalação."
-        log_error "Tente executar: export PATH=\"\$HOME/.local/bin:\$PATH\""
+        log_error "Comando 'comfy' não encontrado no PATH."
         return 1
     fi
     
@@ -162,13 +158,13 @@ install_comfyui() {
 }
 
 # ============================================================================
-# INSTALAÇÃO DO HUGGINGFACE HUB E HF_TRANSFER
+# INSTALAÇÃO DO HUGGINGFACE HUB
 # ============================================================================
 
 install_hf_tools() {
-    log_info "Instalando HuggingFace Hub CLI e hf_transfer..."
+    log_info "Instalando HuggingFace Hub e hf_transfer..."
     
-    python3 -m pip install -U "huggingface_hub[cli]" hf_transfer || {
+    python3 -m pip install -U "huggingface_hub[cli,hf_transfer]" || {
         log_error "Falha ao instalar ferramentas HuggingFace"
         return 1
     }
@@ -181,7 +177,7 @@ install_hf_tools() {
 }
 
 # ============================================================================
-# INSTALAÇÃO DO SAGEATTENTION 2.2.0 (COMPILAÇÃO OTIMIZADA)
+# INSTALAÇÃO DO SAGEATTENTION 2.2.0
 # ============================================================================
 
 install_sageattention() {
@@ -196,7 +192,7 @@ install_sageattention() {
     fi
     
     # Instala dependências necessárias
-    log_info "Instalando dependências do SageAttention (Triton e utilidades)..."
+    log_info "Instalando dependências do SageAttention..."
     python3 -m pip install 'triton>=3.0.0,<3.1.0' packaging ninja || {
         log_error "Falha ao instalar dependências do SageAttention"
         return 1
@@ -223,18 +219,17 @@ install_sageattention() {
         local cuda_capability=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader | head -n1 | tr -d '.')
         if [ -n "$cuda_capability" ]; then
             log_info "Detectada capacidade CUDA: $cuda_capability"
-            # Adiciona a arquitetura detectada se não for Ada
             if [ "$cuda_capability" != "89" ]; then
                 export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;${cuda_capability:0:1}.${cuda_capability:1}"
-                log_info "TORCH_CUDA_ARCH_LIST configurado para: $TORCH_CUDA_ARCH_LIST"
+                log_info "TORCH_CUDA_ARCH_LIST: $TORCH_CUDA_ARCH_LIST"
             fi
         fi
     fi
     
     # Compila e instala
-    log_info "Compilando SageAttention (isso pode demorar 5-10 minutos)..."
+    log_info "Compilando SageAttention (5-10 minutos)..."
     if ! python3 setup.py install >> "$LOG_FILE" 2>&1; then
-        log_error "Falha ao compilar SageAttention. Verifique $LOG_FILE para detalhes."
+        log_error "Falha ao compilar SageAttention. Verifique $LOG_FILE"
         cd "$INSTALL_BASE_DIR"
         return 1
     fi
@@ -244,77 +239,6 @@ install_sageattention() {
     return 0
 }
 
-# ============================================================================
-# FUNÇÃO DE DOWNLOAD COM RETRY E VERIFICAÇÃO
-# ============================================================================
-
-download_file_with_retry() {
-    local url="$1"
-    local dest_dir="$2"
-    local filename="${3:-}"
-    local max_retries=3
-    local retry_count=0
-    
-    # Remove parâmetros de query do nome do arquivo se não foi especificado
-    if [ -z "$filename" ]; then
-        filename=$(basename "${url%%\?*}")
-    fi
-    
-    local dest_file="$dest_dir/$filename"
-    
-    # Cria o diretório de destino se não existir
-    mkdir -p "$dest_dir"
-    
-    while [ $retry_count -lt $max_retries ]; do
-        log_info "Baixando: $filename (tentativa $((retry_count + 1))/$max_retries)"
-        
-        # Usa huggingface-cli se disponível (mais confiável para HF)
-        if command_exists huggingface-cli && [[ "$url" == *"huggingface.co"* ]]; then
-            # Extrai repo e filename do URL do HuggingFace
-            local hf_path=$(echo "$url" | sed -n 's|.*huggingface.co/\([^/]*/[^/]*\)/resolve/main/\(.*\)\?.*|\1 \2|p')
-            if [ -n "$hf_path" ]; then
-                local repo=$(echo "$hf_path" | cut -d' ' -f1)
-                local file_path=$(echo "$hf_path" | cut -d' ' -f2)
-                
-                export HF_HUB_ENABLE_HF_TRANSFER=1
-                if huggingface-cli download "$repo" "$file_path" --local-dir "$dest_dir" --local-dir-use-symlinks False >> "$LOG_FILE" 2>&1; then
-                    # Move o arquivo para o nome correto se necessário
-                    if [ -f "$dest_dir/$file_path" ] && [ "$file_path" != "$filename" ]; then
-                        mv "$dest_dir/$file_path" "$dest_file"
-                    fi
-                    log_info "Download concluído: $filename"
-                    return 0
-                fi
-            fi
-        fi
-        
-        # Fallback para wget/curl
-        if command_exists wget; then
-            if wget -c -O "$dest_file" "$url" >> "$LOG_FILE" 2>&1; then
-                log_info "Download concluído: $filename"
-                return 0
-            fi
-        elif command_exists curl; then
-            if curl -L -C - -o "$dest_file" "$url" >> "$LOG_FILE" 2>&1; then
-                log_info "Download concluído: $filename"
-                return 0
-            fi
-        else
-            log_error "Nenhuma ferramenta de download (wget/curl) encontrada."
-            return 1
-        fi
-        
-        retry_count=$((retry_count + 1))
-        if [ $retry_count -lt $max_retries ]; then
-            log_warning "Download falhou. Tentando novamente em 5 segundos..."
-            sleep 5
-        fi
-    done
-    
-    log_error "Falha ao baixar $filename após $max_retries tentativas"
-    return 1
-}
-
 # ============================================================================
 # DOWNLOAD DO WORKFLOW
 # ============================================================================
@@ -322,19 +246,27 @@ download_file_with_retry() {
 download_workflow() {
     log_info "Baixando o workflow..."
     
-    if download_file_with_retry "$WORKFLOW_URL" "$INSTALL_BASE_DIR" "$WORKFLOW_FILENAME"; then
-        if [ -f "$WORKFLOW_FILE" ]; then
-            log_info "Workflow baixado com sucesso."
-            return 0
-        fi
+    # Usa wget com user-agent para evitar bloqueios
+    if command_exists wget; then
+        wget -U "Mozilla/5.0" -O "$WORKFLOW_FILE" "$WORKFLOW_URL" 2>&1 | tee -a "$LOG_FILE"
+    elif command_exists curl; then
+        curl -L -H "User-Agent: Mozilla/5.0" -o "$WORKFLOW_FILE" "$WORKFLOW_URL" 2>&1 | tee -a "$LOG_FILE"
+    else
+        log_error "wget ou curl não encontrado"
+        return 1
     fi
     
-    log_error "Falha ao baixar o arquivo do workflow."
-    return 1
+    if [ -f "$WORKFLOW_FILE" ] && [ -s "$WORKFLOW_FILE" ]; then
+        log_info "Workflow baixado com sucesso."
+        return 0
+    else
+        log_error "Falha ao baixar o workflow."
+        return 1
+    fi
 }
 
 # ============================================================================
-# DOWNLOAD DE MODELOS WAN 2.2
+# DOWNLOAD DE MODELOS WAN 2.2 (CORRIGIDO)
 # ============================================================================
 
 download_models() {
@@ -351,131 +283,103 @@ download_models() {
     
     for dir in "${dirs[@]}"; do
         mkdir -p "$dir"
-        log_info "Diretório criado: $dir"
+        log_info "Diretório criado/verificado: $dir"
     done
     
-    # Lista de modelos Wan 2.2 - usando huggingface-cli quando possível
-    log_info "Configurando lista de modelos para download..."
+    # Configuração para usar hf
+    export HF_HUB_ENABLE_HF_TRANSFER=1
     
-    # Verifica se huggingface-cli está disponível
-    if command_exists huggingface-cli; then
-        log_info "Usando huggingface-cli para downloads (mais confiável)"
-        export HF_HUB_ENABLE_HF_TRANSFER=1
-        
-        # Download com huggingface-cli
-        local hf_downloads=(
-            # Diffusion Models
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/diffusion_models/wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors:$WORKSPACE_DIR/models/diffusion_models"
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors:$WORKSPACE_DIR/models/diffusion_models"
-            
-            # LoRAs
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/loras/wan2.2_i2v_lightx2v_4steps_lora_v1_high_noise.safetensors:$WORKSPACE_DIR/models/loras"
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/loras/wan2.2_i2v_lightx2v_4steps_lora_v1_low_noise.safetensors:$WORKSPACE_DIR/models/loras"
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/loras/wan2.2_t2v_lightx2v_4steps_lora_v1.1_high_noise.safetensors:$WORKSPACE_DIR/models/loras"
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/loras/wan2.2_t2v_lightx2v_4steps_lora_v1.1_low_noise.safetensors:$WORKSPACE_DIR/models/loras"
-            
-            # Text Encoders
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/text_encoders/umt5_xxl_fp16.safetensors:$WORKSPACE_DIR/models/text_encoders"
-            "Comfy-Org/Wan_2.2_ComfyUI_Repackaged:split_files/text_encoders/umt5_xxl_fp8_e4m3fn_scaled.safetensors:$WORKSPACE_DIR/models/text_encoders"
-            
-            # VAE
-            "Comfy-Org/Wan_2.1_ComfyUI_repackaged:split_files/vae/wan_2.1_vae.safetensors:$WORKSPACE_DIR/models/vae"
-            
-            # CLIP Vision
-            "Comfy-Org/Wan_2.1_ComfyUI_repackaged:split_files/clip_vision/clip_vision_h.safetensors:$WORKSPACE_DIR/models/clip_vision"
-        )
-        
-        local total=${#hf_downloads[@]}
-        local current=0
-        local failed_downloads=()
-        
-        for download_info in "${hf_downloads[@]}"; do
-            IFS=':' read -r repo file dest_dir <<< "$download_info"
-            current=$((current + 1))
-            local filename=$(basename "$file")
-            
-            log_info "[$current/$total] Baixando: $filename"
-            log_info "  Repositório: $repo"
-            log_info "  Arquivo: $file"
-            log_info "  Destino: $dest_dir"
-            
-            # Tenta download com huggingface-cli
-            if huggingface-cli download "$repo" "$file" \
-                --local-dir "$dest_dir" \
-                --local-dir-use-symlinks False \
-                2>&1 | tee -a "$LOG_FILE"; then
-                
-                # Move o arquivo para o local correto se necessário
-                if [ -f "$dest_dir/$file" ]; then
-                    mv "$dest_dir/$file" "$dest_dir/$filename"
-                    log_info "✓ Download concluído: $filename"
-                elif [ -f "$dest_dir/$filename" ]; then
-                    log_info "✓ Download concluído: $filename"
-                else
-                    log_warning "Arquivo baixado mas não encontrado no destino esperado"
-                    failed_downloads+=("$filename")
-                fi
-            else
-                log_warning "Falha ao baixar com huggingface-cli: $filename"
-                failed_downloads+=("$filename")
-            fi
-        done
-        
-    else
-        # Fallback para wget/curl
-        log_info "huggingface-cli não disponível, usando wget/curl"
-        
-        local models=(
-            # URLs completas com destinos
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/diffusion_models/wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors:$WORKSPACE_DIR/models/diffusion_models:wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors:$WORKSPACE_DIR/models/diffusion_models:wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/loras/wan2.2_i2v_lightx2v_4steps_lora_v1_high_noise.safetensors:$WORKSPACE_DIR/models/loras:wan2.2_i2v_lightx2v_4steps_lora_v1_high_noise.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/loras/wan2.2_i2v_lightx2v_4steps_lora_v1_low_noise.safetensors:$WORKSPACE_DIR/models/loras:wan2.2_i2v_lightx2v_4steps_lora_v1_low_noise.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/loras/wan2.2_t2v_lightx2v_4steps_lora_v1.1_high_noise.safetensors:$WORKSPACE_DIR/models/loras:wan2.2_t2v_lightx2v_4steps_lora_v1.1_high_noise.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/loras/wan2.2_t2v_lightx2v_4steps_lora_v1.1_low_noise.safetensors:$WORKSPACE_DIR/models/loras:wan2.2_t2v_lightx2v_4steps_lora_v1.1_low_noise.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/text_encoders/umt5_xxl_fp16.safetensors:$WORKSPACE_DIR/models/text_encoders:umt5_xxl_fp16.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.2_ComfyUI_Repackaged/resolve/main/split_files/text_encoders/umt5_xxl_fp8_e4m3fn_scaled.safetensors:$WORKSPACE_DIR/models/text_encoders:umt5_xxl_fp8_e4m3fn_scaled.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/vae/wan_2.1_vae.safetensors:$WORKSPACE_DIR/models/vae:wan_2.1_vae.safetensors"
-            "https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/clip_vision/clip_vision_h.safetensors:$WORKSPACE_DIR/models/clip_vision:clip_vision_h.safetensors"
-        )
-        
-        local total=${#models[@]}
-        local current=0
-        local failed_downloads=()
-        
-        for model_info in "${models[@]}"; do
-            IFS=':' read -r url dest_dir filename <<< "$model_info"
-            current=$((current + 1))
-            
-            log_info "[$current/$total] Baixando: $filename"
-            
-            if ! download_file_with_retry "$url" "$dest_dir" "$filename"; then
-                failed_downloads+=("$filename")
-                log_warning "Falha ao baixar: $filename"
-            fi
-        done
+    # Lista de modelos organizados por tipo
+    log_info "Baixando modelos de difusão..."
+    
+    # Diffusion Models
+    hf download "Comfy-Org/Wan_2.2_ComfyUI_Repackaged" \
+        "split_files/diffusion_models/wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors" \
+        "split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors" \
+        --local-dir "$WORKSPACE_DIR/models" \
+        --cache-dir "$WORKSPACE_DIR/models/.cache" 2>&1 | tee -a "$LOG_FILE"
+    
+    log_info "Baixando LoRAs..."
+    
+    # LoRAs
+    hf download "Comfy-Org/Wan_2.2_ComfyUI_Repackaged" \
+        "split_files/loras/wan2.2_i2v_lightx2v_4steps_lora_v1_high_noise.safetensors" \
+        "split_files/loras/wan2.2_i2v_lightx2v_4steps_lora_v1_low_noise.safetensors" \
+        "split_files/loras/wan2.2_t2v_lightx2v_4steps_lora_v1.1_high_noise.safetensors" \
+        "split_files/loras/wan2.2_t2v_lightx2v_4steps_lora_v1.1_low_noise.safetensors" \
+        --local-dir "$WORKSPACE_DIR/models" \
+        --cache-dir "$WORKSPACE_DIR/models/.cache" 2>&1 | tee -a "$LOG_FILE"
+    
+    log_info "Baixando Text Encoders..."
+    
+    # Text Encoders
+    hf download "Comfy-Org/Wan_2.2_ComfyUI_Repackaged" \
+        "split_files/text_encoders/umt5_xxl_fp16.safetensors" \
+        "split_files/text_encoders/umt5_xxl_fp8_e4m3fn_scaled.safetensors" \
+        --local-dir "$WORKSPACE_DIR/models" \
+        --cache-dir "$WORKSPACE_DIR/models/.cache" 2>&1 | tee -a "$LOG_FILE"
+    
+    log_info "Baixando VAE..."
+    
+    # VAE
+    hf download "Comfy-Org/Wan_2.1_ComfyUI_repackaged" \
+        "split_files/vae/wan_2.1_vae.safetensors" \
+        --local-dir "$WORKSPACE_DIR/models" \
+        --cache-dir "$WORKSPACE_DIR/models/.cache" 2>&1 | tee -a "$LOG_FILE"
+    
+    log_info "Baixando CLIP Vision..."
+    
+    # CLIP Vision
+    hf download "Comfy-Org/Wan_2.1_ComfyUI_repackaged" \
+        "split_files/clip_vision/clip_vision_h.safetensors" \
+        --local-dir "$WORKSPACE_DIR/models" \
+        --cache-dir "$WORKSPACE_DIR/models/.cache" 2>&1 | tee -a "$LOG_FILE"
+    
+    # Move arquivos dos subdiretórios para os locais corretos
+    log_info "Organizando arquivos de modelos..."
+    
+    # Move diffusion models
+    if [ -d "$WORKSPACE_DIR/models/split_files/diffusion_models" ]; then
+        mv "$WORKSPACE_DIR/models/split_files/diffusion_models/"*.safetensors "$WORKSPACE_DIR/models/diffusion_models/" 2>/dev/null || true
     fi
     
-    # Relatório final
-    log_info "================================================="
-    if [ ${#failed_downloads[@]} -gt 0 ]; then
-        log_warning "Downloads com falha:"
-        for model in "${failed_downloads[@]}"; do
-            log_warning "  ✗ $model"
-        done
-        log_warning "Execute o script novamente ou baixe manualmente os modelos faltando."
-        return 1
-    else
-        log_info "✓ Todos os modelos foram baixados com sucesso!"
+    # Move loras
+    if [ -d "$WORKSPACE_DIR/models/split_files/loras" ]; then
+        mv "$WORKSPACE_DIR/models/split_files/loras/"*.safetensors "$WORKSPACE_DIR/models/loras/" 2>/dev/null || true
+    fi
+    
+    # Move text encoders
+    if [ -d "$WORKSPACE_DIR/models/split_files/text_encoders" ]; then
+        mv "$WORKSPACE_DIR/models/split_files/text_encoders/"*.safetensors "$WORKSPACE_DIR/models/text_encoders/" 2>/dev/null || true
     fi
-    log_info "================================================="
     
-    # Lista arquivos baixados para verificação
+    # Move vae
+    if [ -d "$WORKSPACE_DIR/models/split_files/vae" ]; then
+        mv "$WORKSPACE_DIR/models/split_files/vae/"*.safetensors "$WORKSPACE_DIR/models/vae/" 2>/dev/null || true
+    fi
+    
+    # Move clip vision
+    if [ -d "$WORKSPACE_DIR/models/split_files/clip_vision" ]; then
+        mv "$WORKSPACE_DIR/models/split_files/clip_vision/"*.safetensors "$WORKSPACE_DIR/models/clip_vision/" 2>/dev/null || true
+    fi
+    
+    # Remove diretório split_files vazio
+    rm -rf "$WORKSPACE_DIR/models/split_files" 2>/dev/null || true
+    
+    # Verifica arquivos baixados
     log_info "Verificando modelos baixados:"
     for dir in "${dirs[@]}"; do
         if [ -d "$dir" ]; then
-            local count=$(ls -1 "$dir" 2>/dev/null | wc -l)
-            log_info "  $dir: $count arquivo(s)"
+            local count=$(ls -1 "$dir"/*.safetensors 2>/dev/null | wc -l)
+            log_info "  $(basename $(dirname "$dir"))/$(basename "$dir"): $count arquivo(s)"
+            
+            # Lista os arquivos
+            for file in "$dir"/*.safetensors; do
+                if [ -f "$file" ]; then
+                    local size=$(du -h "$file" | cut -f1)
+                    log_info "    - $(basename "$file") ($size)"
+                fi
+            done
         fi
     done
     
@@ -483,48 +387,81 @@ download_models() {
 }
 
 # ============================================================================
-# INSTALAÇÃO DE CUSTOM NODES E DEPENDÊNCIAS
+# INSTALAÇÃO MANUAL DE CUSTOM NODES
 # ============================================================================
 
-install_custom_nodes_and_deps() {
-    log_info "Instalando custom nodes e dependências..."
+install_custom_nodes_manually() {
+    log_info "Instalando custom nodes manualmente..."
     
-    # Instala dependências do workflow se existir
-    if [ -f "$WORKFLOW_FILE" ]; then
-        log_info "Instalando dependências do workflow..."
-        comfy --workspace "$WORKSPACE_DIR" node install-deps --workflow="$WORKFLOW_FILE" >> "$LOG_FILE" 2>&1 || {
-            log_warning "Algumas dependências do workflow podem ter falhado ao instalar."
-        }
-    fi
+    local custom_nodes_dir="$WORKSPACE_DIR/custom_nodes"
     
-    # Lista de custom nodes
+    # Lista completa de custom nodes (sem duplicatas)
     local custom_nodes=(
-        "https://github.com/LAOGOU-666/Comfyui-Memory_Cleanup"
-        "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes"
+        "https://github.com/kijai/ComfyUI-WanVideoWrapper"
+        "https://github.com/kijai/ComfyUI-MelBandRoFormer"
         "https://github.com/kijai/ComfyUI-KJNodes"
-        "https://github.com/kijai/ComfyUI-WanVideoWrapper.git"
+        "https://github.com/kijai/ComfyUI-Florence2"
+        "https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite"
+        "https://github.com/Fannovel16/ComfyUI-Frame-Interpolation"
+        "https://github.com/Artificial-Sweetener/comfyui-WhiteRabbit"
+        "https://github.com/rdancer/ComfyUI_Florence2SAM2"
+        "https://github.com/1038lab/ComfyUI-RMBG"
+        "https://github.com/fpgaminer/joycaption_comfyui"
+        "https://github.com/MieMieeeee/ComfyUI-CaptionThis"
+        "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes"
+        "https://github.com/LAOGOU-666/Comfyui-Memory_Cleanup"
+        "https://github.com/chrisgoringe/cg-use-everywhere"
+        "https://github.com/pythongosssss/ComfyUI-Custom-Scripts"
+        "https://github.com/chibiace/ComfyUI-Chibi-Nodes"
+        "https://github.com/ptmaster/comfyui-audio-speed"
+        "https://github.com/filliptm/ComfyUI_Fill-Nodes"
+        "https://github.com/justUmen/Bjornulf_custom_nodes"
+        "https://github.com/raindrop313/ComfyUI-WanVideoStartEndFrames"
+        "https://github.com/ShmuelRonen/ComfyUI_wav2lip"
+        "https://github.com/crystian/ComfyUI-Crystools"
+        "https://github.com/bash-j/mikey_nodes"
     )
     
-    local failed_nodes=()
+    cd "$custom_nodes_dir"
     
     for node_url in "${custom_nodes[@]}"; do
-        log_info "Instalando custom node: $(basename "$node_url" .git)"
-        if ! comfy --workspace "$WORKSPACE_DIR" node install "$node_url" >> "$LOG_FILE" 2>&1; then
-            failed_nodes+=("$node_url")
-            log_warning "Falha ao instalar: $(basename "$node_url" .git)"
+        local node_name=$(basename "$node_url" .git)
+        log_info "Instalando: $node_name"
+        
+        # Remove instalação anterior se existir
+        if [ -d "$node_name" ]; then
+            log_info "  Removendo versão anterior..."
+            rm -rf "$node_name"
+        fi
+        
+        # Clona o repositório
+        if git clone --depth 1 "$node_url" "$node_name" 2>&1 | tee -a "$LOG_FILE"; then
+            log_info "  Clone concluído: $node_name"
+            
+            # Instala requirements se existir
+            if [ -f "$node_name/requirements.txt" ]; then
+                log_info "  Instalando dependências de $node_name..."
+                python3 -m pip install -r "$node_name/requirements.txt" 2>&1 | tee -a "$LOG_FILE" || {
+                    log_warning "  Algumas dependências de $node_name falharam"
+                }
+            fi
+            
+            # Executa install.py se existir
+            if [ -f "$node_name/install.py" ]; then
+                log_info "  Executando install.py de $node_name..."
+                cd "$node_name"
+                python3 install.py 2>&1 | tee -a "$LOG_FILE" || {
+                    log_warning "  install.py de $node_name retornou erro"
+                }
+                cd ..
+            fi
+        else
+            log_warning "  Falha ao clonar: $node_name"
         fi
     done
     
-    if [ ${#failed_nodes[@]} -gt 0 ]; then
-        log_warning "Os seguintes custom nodes falharam:"
-        for node in "${failed_nodes[@]}"; do
-            log_warning "  - $node"
-        done
-    else
-        log_info "Todos os custom nodes foram instalados com sucesso!"
-    fi
-    
-    return 0
+    cd "$INSTALL_BASE_DIR"
+    log_info "Instalação de custom nodes concluída!"
 }
 
 # ============================================================================
@@ -546,25 +483,49 @@ verify_installation() {
         issues+=("Arquivo de workflow não encontrado")
     fi
     
-    # Verifica pelo menos um modelo
-    if [ -z "$(ls -A "$WORKSPACE_DIR/models/diffusion_models" 2>/dev/null)" ]; then
-        issues+=("Nenhum modelo de difusão encontrado")
-    fi
+    # Verifica modelos em cada diretório
+    log_info "Verificando modelos instalados:"
+    
+    local expected_models=(
+        "diffusion_models:2"
+        "loras:4"
+        "text_encoders:2"
+        "vae:1"
+        "clip_vision:1"
+    )
+    
+    for expectation in "${expected_models[@]}"; do
+        IFS=':' read -r dir_name expected_count <<< "$expectation"
+        local dir="$WORKSPACE_DIR/models/$dir_name"
+        
+        if [ -d "$dir" ]; then
+            local count=$(ls -1 "$dir"/*.safetensors 2>/dev/null | wc -l)
+            log_info "  $dir_name: $count arquivo(s) (esperado: $expected_count)"
+            
+            if [ $count -lt $expected_count ]; then
+                issues+=("Faltam modelos em $dir_name ($count/$expected_count)")
+            fi
+        else
+            issues+=("Diretório não existe: $dir_name")
+        fi
+    done
     
     # Verifica SageAttention
     if ! python3 -c "import sageattention" 2>/dev/null; then
         log_warning "SageAttention não está instalado (opcional)"
+    else
+        log_info "✓ SageAttention instalado corretamente"
     fi
     
     if [ ${#issues[@]} -gt 0 ]; then
         log_error "Problemas encontrados na instalação:"
         for issue in "${issues[@]}"; do
-            log_error "  - $issue"
+            log_error "  ✗ $issue"
         done
         return 1
     fi
     
-    log_info "Instalação verificada com sucesso!"
+    log_info "✓ Instalação verificada com sucesso!"
     return 0
 }
 
@@ -583,8 +544,8 @@ launch_comfyui() {
     log_info "  Workflow: $WORKFLOW_FILE"
     log_info ""
     log_info "Para usar o workflow:"
-    log_info "  1. Acesse http://127.0.0.1:8818 no seu navegador"
-    log_info "  2. Arraste e solte o arquivo '$WORKFLOW_FILENAME' na interface"
+    log_info "  1. Acesse http://127.0.0.1:8818"
+    log_info "  2. Arraste o arquivo '$WORKFLOW_FILENAME' na interface"
     log_info ""
     log_info "Pressione Ctrl+C para parar o servidor"
     log_info "================================================="
@@ -614,7 +575,7 @@ main() {
     echo "=== Instalação ComfyUI iniciada em $(date) ===" > "$LOG_FILE"
     
     log_info "================================================="
-    log_info "Instalador ComfyUI com Wan 2.2"
+    log_info "Instalador ComfyUI com Wan 2.2 - v3.0"
     log_info "================================================="
     
     # Etapa 1: Verificar requisitos
@@ -635,29 +596,27 @@ main() {
     # Etapa 6: Baixar workflow
     download_workflow || log_warning "Workflow não foi baixado"
     
-    # Etapa 7: Instalar custom nodes
-    install_custom_nodes_and_deps
+    # Etapa 7: Instalar custom nodes manualmente
+    install_custom_nodes_manually
     
-    # Etapa 8: PROCESSOS PARALELOS - SageAttention e Downloads de Modelos
+    # Etapa 8: PROCESSOS PARALELOS - SageAttention e Downloads
     log_info "================================================="
     log_info "Iniciando processos paralelos:"
-    log_info "  1. Instalação do SageAttention (5-10 minutos)"
-    log_info "  2. Download dos modelos Wan 2.2 (vários GB)"
+    log_info "  1. Instalação do SageAttention (5-10 min)"
+    log_info "  2. Download dos modelos Wan 2.2"
     log_info "================================================="
     
     # Inicia instalação do SageAttention em background
-    log_info "Iniciando instalação do SageAttention em background..."
     install_sageattention &
     local sage_pid=$!
     
     # Inicia downloads dos modelos em background
-    log_info "Iniciando downloads dos modelos em background..."
     download_models &
     local models_pid=$!
     
     # Monitor de progresso
     log_info "Aguardando conclusão dos processos paralelos..."
-    log_info "Você pode monitorar o progresso em: $LOG_FILE"
+    log_info "Monitore o progresso em: $LOG_FILE"
     
     local both_done=false
     local sage_done=false
@@ -696,22 +655,23 @@ main() {
         if [ "$sage_done" = true ] && [ "$models_done" = true ]; then
             both_done=true
         else
-            # Mostra status a cada 30 segundos
-            sleep 30
+            sleep 10
             if [ "$sage_done" = false ] && ps -p $sage_pid > /dev/null 2>&1; then
-                log_info "... SageAttention ainda compilando..."
+                echo -n "."
             fi
             if [ "$models_done" = false ] && ps -p $models_pid > /dev/null 2>&1; then
-                log_info "... Downloads ainda em progresso..."
+                echo -n "."
             fi
         fi
     done
     
+    echo ""
     log_info "Todos os processos paralelos foram concluídos!"
     
     # Etapa 9: Verificar instalação
     verify_installation || {
         log_error "A instalação teve problemas. Verifique o log em: $LOG_FILE"
+        log_error "Você pode tentar executar o script novamente."
         exit 1
     }