Update WANcomfyui_installer_fixed.sh

WANcomfyui_installer_fixed.sh

@@ -2,7 +2,7 @@
 
 # ============================================================================
 # Instalador Otimizado para ComfyUI com Wan 2.2
-# Versão 3.0 - Corrigida com instalação manual de custom nodes
+# Versão 3.1 - Com instalação do PyTorch CUDA 12.8
 # ============================================================================
 
 set -e
@@ -183,6 +183,20 @@ install_hf_tools() {
 install_sageattention() {
     log_info "Instalando SageAttention 2.2.0 com 2++ (compilação otimizada)..."
     
+    # =========================================================================
+    # INSTALAÇÃO DO PYTORCH COM CUDA 12.8
+    # =========================================================================
+    log_info "Instalando PyTorch 2.7.0 com CUDA 12.8..."
+    
+    python3 -m pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128 || {
+        log_error "Falha ao instalar PyTorch com CUDA 12.8"
+        return 1
+    }
+    
+    # Verifica instalação do PyTorch
+    log_info "Verificando instalação do PyTorch..."
+    python3 -c "import torch; print(f'PyTorch version: {torch.__version__}'); print(f'CUDA available: {torch.cuda.is_available()}'); print(f'CUDA version: {torch.version.cuda if torch.cuda.is_available() else \"N/A\"}')" 2>&1 | tee -a "$LOG_FILE"
+    
     local sage_dir="$INSTALL_BASE_DIR/SageAttention"
     
     # Remove instalação anterior se existir
@@ -510,6 +524,14 @@ verify_installation() {
         fi
     done
     
+    # Verifica PyTorch com CUDA
+    log_info "Verificando PyTorch e CUDA:"
+    if python3 -c "import torch; assert torch.cuda.is_available()" 2>/dev/null; then
+        log_info "✓ PyTorch com CUDA instalado corretamente"
+    else
+        log_warning "PyTorch instalado mas CUDA não está disponível"
+    fi
+    
     # Verifica SageAttention
     if ! python3 -c "import sageattention" 2>/dev/null; then
         log_warning "SageAttention não está instalado (opcional)"
@@ -575,7 +597,8 @@ main() {
     echo "=== Instalação ComfyUI iniciada em $(date) ===" > "$LOG_FILE"
     
     log_info "================================================="
-    log_info "Instalador ComfyUI com Wan 2.2 - v3.0"
+    log_info "Instalador ComfyUI com Wan 2.2 - v3.1"
+    log_info "Com PyTorch 2.7.0 + CUDA 12.8"
     log_info "================================================="
     
     # Etapa 1: Verificar requisitos
@@ -602,7 +625,7 @@ main() {
     # Etapa 8: PROCESSOS PARALELOS - SageAttention e Downloads
     log_info "================================================="
     log_info "Iniciando processos paralelos:"
-    log_info "  1. Instalação do SageAttention (5-10 min)"
+    log_info "  1. Instalação do PyTorch + SageAttention (10-15 min)"
     log_info "  2. Download dos modelos Wan 2.2"
     log_info "================================================="
     
@@ -630,9 +653,9 @@ main() {
                 local sage_exit=$?
                 sage_done=true
                 if [ $sage_exit -eq 0 ]; then
-                    log_info "✓ SageAttention instalado com sucesso!"
+                    log_info "✓ PyTorch + SageAttention instalado com sucesso!"
                 else
-                    log_warning "⚠ SageAttention pode não ter sido instalado corretamente"
+                    log_warning "⚠ PyTorch/SageAttention pode não ter sido instalado corretamente"
                 fi
             fi
         fi