Update setup_comfy_wan222.sh

setup_comfy_wan222.sh

@@ -316,7 +316,7 @@ log_info "========================================="
 echo ""
 
 # [1] Verificar e instalar dependências
-log_info "[1/7] Verificando e instalando dependências..."
+log_info "[1/8] Verificando e instalando dependências..."
 
 # Instalar aria2c do sistema se não existir
 if ! command_exists aria2c; then
@@ -333,7 +333,7 @@ done
 log_success "Dependências OK"
 
 # [2] Atualizar pip e instalar ferramentas
-log_info "[2/7] Preparando ferramentas Python..."
+log_info "[2/8] Preparando ferramentas Python..."
 python3 -m pip install --upgrade pip wheel setuptools -q
 
 # Instalar huggingface_hub com hf_transfer
@@ -346,8 +346,13 @@ fi
 
 log_success "Ferramentas instaladas"
 
-# [3] Instalar ComfyUI
-log_info "[3/7] Instalando ComfyUI..."
+# [3] Instalar PyTorch com CUDA 12.8
+log_info "[3/8] Instalando PyTorch com CUDA 12.8..."
+python3 -m pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
+log_success "PyTorch com CUDA 12.8 instalado"
+
+# [4] Instalar ComfyUI
+log_info "[4/8] Instalando ComfyUI..."
 if [ -f "$COMFY_DIR/main.py" ]; then
     log_warn "ComfyUI já existe, pulando instalação base"
 else
@@ -361,31 +366,24 @@ else
 fi
 log_success "ComfyUI instalado em: $COMFY_DIR"
 
-# [4] Instalar SageAttention
-log_info "[4/7] Instalando SageAttention..."
-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"
+# [5] Instalar SageAttention (wheel pré-compilada)
+log_info "[5/8] Instalando SageAttention..."
+if command_exists nvcc || python3 -c "import torch; print(torch.cuda.is_available())" 2>/dev/null | grep -q "True"; then
+    log_info "Instalando SageAttention pré-compilada..."
+    python3 -m pip install "https://huggingface.co/adbrasi/comfywheel/resolve/main/sageattention-2.2.0-cp311-cp311-linux_x86_64.whl" -q
     
-    if git clone --depth 1 https://github.com/thu-ml/SageAttention.git "$SAGE_DIR" 2>/dev/null; then
-        cd "$SAGE_DIR"
-        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"
+    # Verificar instalação
+    if python3 -c "import sageattention" 2>/dev/null; then
+        log_success "SageAttention instalado com sucesso"
     else
-        log_warn "Não foi possível clonar SageAttention, continuando sem ele"
+        log_warn "SageAttention instalado mas não pôde ser importado"
     fi
 else
     log_warn "CUDA não detectado, pulando SageAttention"
 fi
 
-# [5] Baixar modelos
-log_info "[5/7] Baixando modelos..."
+# [6] Baixar modelos
+log_info "[6/8] Baixando modelos..."
 mkdir -p "$MODELS_DIR"/{diffusion_models,loras,vae,text_encoders,clip_vision,controlnet,upscale_models}
 
 # Verificar ferramentas de download
@@ -402,8 +400,8 @@ else
     log_warn "Alguns downloads falharam, mas continuando..."
 fi
 
-# [6] Instalar custom nodes
-log_info "[6/7] Instalando custom nodes..."
+# [7] Instalar custom nodes
+log_info "[7/8] Instalando custom nodes..."
 CN_DIR="$COMFY_DIR/custom_nodes"
 mkdir -p "$CN_DIR"
 
@@ -429,9 +427,9 @@ wait
 
 log_success "Custom nodes instalados"
 
-# [7] Verificação final
+# [8] Verificação final
 echo ""
-log_info "[7/7] Verificando instalação..."
+log_info "[8/8] Verificando instalação..."
 
 # Verificar arquivos principais
 check_critical_files() {