Update setup_comfyui_wan22.sh

setup_comfyui_wan22.sh

@@ -1,6 +1,6 @@
 #!/usr/bin/env bash
-# setup_comfyui_wan22_final.sh
-# Script final corrigido para instalação do ComfyUI com Wan 2.2
+# setup_comfyui_wan22_verbose.sh
+# Script com logs completos para instalação do ComfyUI com Wan 2.2
 # Requisitos: Linux + Python3 + git + (opcional) NVCC/CUDA para SageAttention
 
 set -Eeuo pipefail
@@ -13,12 +13,16 @@ RED='\033[0;31m'
 GREEN='\033[0;32m'
 YELLOW='\033[1;33m'
 BLUE='\033[0;34m'
+CYAN='\033[0;36m'
+MAGENTA='\033[0;35m'
 NC='\033[0m' # No Color
 
 log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
 log_success() { echo -e "${GREEN}[✓]${NC} $1"; }
 log_warn() { echo -e "${YELLOW}[!]${NC} $1"; }
 log_error() { echo -e "${RED}[✗]${NC} $1"; }
+log_debug() { echo -e "${CYAN}[DEBUG]${NC} $1"; }
+log_cmd() { echo -e "${MAGENTA}[CMD]${NC} $1"; }
 
 # -----------------------------
 # Configuráveis
@@ -40,26 +44,105 @@ CIVITAI_TOKEN="${CIVITAI_TOKEN:-4fcb2834969399006a736ee402b061e5}"
 export HF_HUB_ENABLE_HF_TRANSFER=1
 export HF_TRANSFER_CONCURRENCY=16  # Aumentado para melhor performance
 
+# -----------------------------
+# Detectar capacidades do CUDA
+# -----------------------------
+detect_cuda_capabilities() {
+    if command -v nvidia-smi >/dev/null 2>&1; then
+        log_info "Detectando capacidades do CUDA..."
+        
+        # Obter informações da GPU
+        GPU_NAME=$(nvidia-smi --query-gpu=name --format=csv,noheader | head -n1)
+        GPU_MEMORY=$(nvidia-smi --query-gpu=memory.total --format=csv,noheader,nounits | head -n1)
+        GPU_COUNT=$(nvidia-smi --query-gpu=count --format=csv,noheader | wc -l)
+        
+        log_debug "GPU: $GPU_NAME ($GPU_MEMORY MB) x$GPU_COUNT"
+        
+        # Detectar compute capability
+        if command -v nvcc >/dev/null 2>&1; then
+            CUDA_VERSION=$(nvcc --version | grep "release" | awk '{print $6}' | cut -d',' -f1)
+            log_debug "CUDA Version: $CUDA_VERSION"
+            
+            # Obter compute capability via nvidia-ml-py se disponível
+            COMPUTE_CAP=$(python3 -c "
+try:
+    import pynvml
+    pynvml.nvmlInit()
+    handle = pynvml.nvmlDeviceGetHandleByIndex(0)
+    major, minor = pynvml.nvmlDeviceGetCudaComputeCapability(handle)
+    print(f'{major}.{minor}')
+except:
+    print('unknown')
+" 2>/dev/null || echo "unknown")
+            
+            if [ "$COMPUTE_CAP" != "unknown" ]; then
+                log_debug "Compute Capability: $COMPUTE_CAP"
+            fi
+            
+            # Determinar MAX_JOBS baseado na GPU e memória
+            # RTX 4090/A100: 128 SMs -> pode usar 32 jobs
+            # RTX 3090: 82 SMs -> pode usar 24 jobs
+            # RTX 3080: 68 SMs -> pode usar 16 jobs
+            # Regra geral: ~4 jobs por GB de VRAM, máximo 32
+            
+            if [ "$GPU_MEMORY" -gt 0 ]; then
+                SUGGESTED_JOBS=$((GPU_MEMORY / 1024 / 4))  # 4 jobs por GB
+                if [ $SUGGESTED_JOBS -gt 32 ]; then
+                    SUGGESTED_JOBS=32  # Limite do CUDA
+                elif [ $SUGGESTED_JOBS -lt 4 ]; then
+                    SUGGESTED_JOBS=4   # Mínimo razoável
+                fi
+                
+                # Para GPUs conhecidas, usar valores otimizados
+                if [[ "$GPU_NAME" == *"4090"* ]] || [[ "$GPU_NAME" == *"A100"* ]]; then
+                    SUGGESTED_JOBS=32
+                    SUGGESTED_THREADS=16
+                elif [[ "$GPU_NAME" == *"3090"* ]] || [[ "$GPU_NAME" == *"A6000"* ]]; then
+                    SUGGESTED_JOBS=24
+                    SUGGESTED_THREADS=12
+                elif [[ "$GPU_NAME" == *"3080"* ]] || [[ "$GPU_NAME" == *"A5000"* ]]; then
+                    SUGGESTED_JOBS=16
+                    SUGGESTED_THREADS=8
+                else
+                    SUGGESTED_THREADS=$((SUGGESTED_JOBS / 2))
+                fi
+                
+                export MAX_JOBS=$SUGGESTED_JOBS
+                export NVCC_APPEND_FLAGS="--threads $SUGGESTED_THREADS"
+                export EXT_PARALLEL=$((SUGGESTED_JOBS / 2))
+                
+                log_success "Configuração CUDA otimizada:"
+                log_debug "  MAX_JOBS=$MAX_JOBS"
+                log_debug "  NVCC threads=$SUGGESTED_THREADS"
+                log_debug "  EXT_PARALLEL=$EXT_PARALLEL"
+            fi
+        fi
+    else
+        log_warn "nvidia-smi não encontrado, usando configuração padrão"
+    fi
+}
+
 # Detecção de CPU cores para compilação
 CPU_CORES=$(nproc 2>/dev/null || sysctl -n hw.ncpu 2>/dev/null || echo 4)
+log_debug "CPU Cores detectados: $CPU_CORES"
 
-# Otimização para máquinas com muitos cores (como 112 cores)
+# Configuração inicial baseada em CPU (será sobrescrita pela detecção CUDA se disponível)
 if [ $CPU_CORES -gt 32 ]; then
-    # Para máquinas high-end, usar configurações agressivas
-    export MAX_JOBS=32  # Limite do CUDA
-    export EXT_PARALLEL=16  # Aumentado
-    export NVCC_APPEND_FLAGS="--threads 16"  # Aumentado
+    export MAX_JOBS=${MAX_JOBS:-32}
+    export EXT_PARALLEL=${EXT_PARALLEL:-16}
+    export NVCC_APPEND_FLAGS=${NVCC_APPEND_FLAGS:-"--threads 16"}
     COMPILE_JOBS=32
-    log_info "Detectado sistema high-end: $CPU_CORES cores, usando MAX_JOBS=$MAX_JOBS"
 else
-    # Para máquinas normais
     HALF_CORES=$((CPU_CORES / 2))
-    export MAX_JOBS=$HALF_CORES
-    export EXT_PARALLEL=4
-    export NVCC_APPEND_FLAGS="--threads 8"
+    export MAX_JOBS=${MAX_JOBS:-$HALF_CORES}
+    export EXT_PARALLEL=${EXT_PARALLEL:-4}
+    export NVCC_APPEND_FLAGS=${NVCC_APPEND_FLAGS:-"--threads 8"}
     COMPILE_JOBS=$HALF_CORES
 fi
 
+# Detectar e otimizar para CUDA
+detect_cuda_capabilities
+
 export UV_SYSTEM_PYTHON=1
 export PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"
 
@@ -97,14 +180,17 @@ clone_or_update() {
     
     if [ -d "$dest/.git" ]; then
         log_info "Atualizando $repo_name..."
-        git -C "$dest" pull --ff-only --quiet || true
+        log_cmd "git -C \"$dest\" pull --ff-only"
+        git -C "$dest" pull --ff-only || true
     else
         log_info "Clonando $repo_name..."
-        git clone --recursive --depth 1 "$url" "$dest" --quiet
+        log_cmd "git clone --recursive --depth 1 \"$url\" \"$dest\""
+        git clone --recursive --depth 1 "$url" "$dest"
     fi
     
     if [ -f "$dest/requirements.txt" ]; then
-        python3 -m pip install --quiet --no-warn-script-location -r "$dest/requirements.txt" 2>/dev/null || true
+        log_debug "Instalando requirements para $repo_name..."
+        python3 -m pip install --no-warn-script-location -r "$dest/requirements.txt"
     fi
 }
 
@@ -118,11 +204,13 @@ hf_download_with_fallback() {
     # Tentar primeiro com HF Transfer
     if python3 -c "import hf_transfer" 2>/dev/null; then
         log_info "  → Usando HF Transfer (rápido) para $repo..."
+        log_cmd "HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download \"$repo\" ..."
+        
         HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo" \
             "${files[@]}" \
             --local-dir "$output_dir" \
             --local-dir-use-symlinks False \
-            --resume-download 2>/dev/null || {
+            --resume-download || {
             log_warn "  HF Transfer falhou, tentando método padrão..."
             HF_HUB_ENABLE_HF_TRANSFER=0 huggingface-cli download "$repo" \
                 "${files[@]}" \
@@ -145,21 +233,25 @@ hf_download_with_fallback() {
 # -----------------------------
 echo ""
 log_info "========================================="
-log_info "  ComfyUI + Wan 2.2 Setup Final"
-log_info "  CPU Cores: $CPU_CORES | Compile Jobs: $COMPILE_JOBS"
+log_info "  ComfyUI + Wan 2.2 Setup (Verbose)"
+log_info "  CPU: $CPU_CORES cores | Compile Jobs: $COMPILE_JOBS"
+log_info "  CUDA MAX_JOBS: ${MAX_JOBS:-não detectado}"
 log_info "  ComfyUI Dir: $COMFY_DIR"
 log_info "========================================="
 echo ""
 
 # [1/8] Atualizar pip e instalar ferramentas essenciais
 log_info "[1/8] Atualizando pip e instalando ferramentas..."
-python3 -m pip install --quiet --upgrade pip wheel setuptools >/dev/null 2>&1
+log_cmd "python3 -m pip install --upgrade pip wheel setuptools"
+python3 -m pip install --upgrade pip wheel setuptools
 
 # Instalar HF Transfer primeiro para downloads mais rápidos
-python3 -m pip install --quiet --upgrade hf_transfer >/dev/null 2>&1 || true
+log_cmd "python3 -m pip install --upgrade hf_transfer"
+python3 -m pip install --upgrade hf_transfer || true
 
 # Depois instalar o resto
-python3 -m pip install --quiet --upgrade comfy-cli "huggingface_hub>=0.23.0" aria2 >/dev/null 2>&1
+log_cmd "python3 -m pip install --upgrade comfy-cli huggingface_hub>=0.23.0 aria2"
+python3 -m pip install --upgrade comfy-cli "huggingface_hub>=0.23.0" aria2
 log_success "Ferramentas instaladas"
 
 # [2/8] Instalar ComfyUI
@@ -175,6 +267,7 @@ else
     [ -d "$COMFY_DIR" ] && rm -rf "$COMFY_DIR"
     
     # Instala ComfyUI (sempre vai para /root/comfy/ComfyUI)
+    log_cmd "comfy --skip-prompt install --fast-deps --nvidia"
     comfy --skip-prompt install --fast-deps --nvidia
     
     # Verifica se instalou no local correto
@@ -221,10 +314,11 @@ log_info "[3/8] Iniciando download dos modelos Wan 2.2..."
         "split_files/text_encoders/umt5_xxl_fp16.safetensors"
     
     # Copiar para destino
-    cp -f "$TMPDL/wan22/split_files/diffusion_models/"*.safetensors "$MODELS_DIR/diffusion_models/" 2>/dev/null || true
-    cp -f "$TMPDL/wan22/split_files/loras/"*.safetensors "$MODELS_DIR/loras/" 2>/dev/null || true
-    cp -f "$TMPDL/wan22/split_files/vae/"*.safetensors "$MODELS_DIR/vae/" 2>/dev/null || true
-    cp -f "$TMPDL/wan22/split_files/text_encoders/"*.safetensors "$MODELS_DIR/text_encoders/" 2>/dev/null || true
+    log_debug "Copiando modelos Wan 2.2 para $MODELS_DIR..."
+    cp -fv "$TMPDL/wan22/split_files/diffusion_models/"*.safetensors "$MODELS_DIR/diffusion_models/" 2>/dev/null || true
+    cp -fv "$TMPDL/wan22/split_files/loras/"*.safetensors "$MODELS_DIR/loras/" 2>/dev/null || true
+    cp -fv "$TMPDL/wan22/split_files/vae/"*.safetensors "$MODELS_DIR/vae/" 2>/dev/null || true
+    cp -fv "$TMPDL/wan22/split_files/text_encoders/"*.safetensors "$MODELS_DIR/text_encoders/" 2>/dev/null || true
     
     log_success "  Modelos Wan 2.2 baixados"
 ) &
@@ -238,7 +332,8 @@ log_info "[4/8] Baixando Wan 2.1 clip vision..."
     hf_download_with_fallback "$REPO_21" \
         "split_files/clip_vision/clip_vision_h.safetensors"
     
-    cp -f "$TMPDL/wan21/split_files/clip_vision/"*.safetensors "$MODELS_DIR/clip_vision/" 2>/dev/null || true
+    log_debug "Copiando clip vision para $MODELS_DIR/clip_vision..."
+    cp -fv "$TMPDL/wan21/split_files/clip_vision/"*.safetensors "$MODELS_DIR/clip_vision/" 2>/dev/null || true
     
     log_success "  Clip vision baixado"
 ) &
@@ -252,7 +347,8 @@ log_info "[5/8] Baixando ControlNet Union..."
     hf_download_with_fallback "$REPO_CN" \
         "diffusion_pytorch_model_promax.safetensors"
     
-    cp -f "$TMPDL/controlnet/diffusion_pytorch_model_promax.safetensors" \
+    log_debug "Copiando ControlNet para $MODELS_DIR/controlnet..."
+    cp -fv "$TMPDL/controlnet/diffusion_pytorch_model_promax.safetensors" \
         "$MODELS_DIR/controlnet/controlnet-union.safetensors"
     
     log_success "  ControlNet baixado"
@@ -269,24 +365,47 @@ log_info "[6/8] Baixando modelo do Civitai..."
     # URL correta completa
     CIVITAI_URL="https://civitai.com/api/download/models/2122278?type=Model&format=SafeTensor&size=pruned&fp=fp16&token=${CIVITAI_TOKEN}"
     
+    log_debug "Baixando de: $CIVITAI_URL"
+    log_cmd "wget --content-disposition \"$CIVITAI_URL\""
+    
     # Tentar com wget primeiro (melhor para nomes de arquivo)
+    # Não suprimir output para ver o progresso
     if ! wget --content-disposition \
               --tries=3 \
               --timeout=60 \
               --continue \
-              --progress=bar:force:noscroll \
-              "$CIVITAI_URL" 2>&1 | grep --line-buffered "%" | sed -u -e "s/^/    /"; then
+              --progress=bar:force \
+              "$CIVITAI_URL"; then
         
         log_warn "  Wget falhou, tentando com curl..."
+        
+        # Tentar obter o nome do arquivo do header primeiro
+        FILENAME=$(curl -sI -L "$CIVITAI_URL" | grep -i "content-disposition" | sed -n 's/.*filename="\([^"]*\)".*/\1/p')
+        
+        if [ -z "$FILENAME" ]; then
+            # Se não conseguir detectar, usar nome genérico
+            FILENAME="civitai_model_$(date +%s).safetensors"
+            log_warn "Não foi possível detectar nome do arquivo, usando: $FILENAME"
+        else
+            log_debug "Nome do arquivo detectado: $FILENAME"
+        fi
+        
+        log_cmd "curl -L \"$CIVITAI_URL\" -o \"$FILENAME\""
         curl -L \
              --retry 3 \
              --connect-timeout 30 \
              --progress-bar \
-             -o "flux_unleashed_v2.safetensors" \
+             -o "$FILENAME" \
              "$CIVITAI_URL"
     fi
     
-    log_success "  Modelo Civitai baixado"
+    # Verificar se algum arquivo foi baixado
+    DOWNLOADED_FILE=$(ls -t *.safetensors 2>/dev/null | head -n1)
+    if [ -n "$DOWNLOADED_FILE" ]; then
+        log_success "  Modelo Civitai baixado: $DOWNLOADED_FILE"
+    else
+        log_error "  Falha ao baixar modelo Civitai"
+    fi
 ) &
 CIVITAI_PID=$!
 
@@ -301,23 +420,40 @@ log_info "[7/8] Preparando SageAttention (paralelo com downloads)..."
         log_info "  Usando configuração otimizada: MAX_JOBS=$MAX_JOBS, EXT_PARALLEL=$EXT_PARALLEL"
         
         # Instalar triton primeiro
-        python3 -m pip install --quiet "triton>=3.0.0" 2>/dev/null || true
+        log_cmd "python3 -m pip install \"triton>=3.0.0\""
+        python3 -m pip install "triton>=3.0.0"
+        
+        # Instalar pynvml para melhor detecção de GPU
+        log_cmd "python3 -m pip install pynvml"
+        python3 -m pip install pynvml || true
         
         # Clonar SageAttention
         SAGE_DIR="$TMPDL/SageAttention"
         if [ ! -d "$SAGE_DIR" ]; then
-            git clone --depth 1 https://github.com/thu-ml/SageAttention.git "$SAGE_DIR" --quiet
+            log_cmd "git clone --depth 1 https://github.com/thu-ml/SageAttention.git \"$SAGE_DIR\""
+            git clone --depth 1 https://github.com/thu-ml/SageAttention.git "$SAGE_DIR"
         fi
         
         cd "$SAGE_DIR"
         
-        # Compilar com configurações otimizadas para muitos cores
-        log_info "  Compilando SageAttention com $MAX_JOBS jobs paralelos..."
-        python3 setup.py install 2>&1 | grep -E "(Building|Compiling|Installing)" || true
+        # Compilar com configurações otimizadas
+        log_info "  Compilando SageAttention com MAX_JOBS=$MAX_JOBS..."
+        log_debug "  NVCC flags: $NVCC_APPEND_FLAGS"
+        
+        # Mostrar output completo da compilação
+        log_cmd "python3 setup.py install"
+        python3 setup.py install
         
         # Testar se instalou
         if python3 -c "import sageattention" 2>/dev/null; then
             log_success "  SageAttention instalado com sucesso"
+            
+            # Mostrar informações sobre a instalação
+            python3 -c "
+import sageattention
+print(f'  SageAttention version: {sageattention.__version__ if hasattr(sageattention, \"__version__\") else \"unknown\"}')
+print(f'  SageAttention path: {sageattention.__file__}')
+" 2>/dev/null || true
         else
             log_warn "  SageAttention não foi instalado corretamente"
         fi
@@ -370,6 +506,7 @@ log_info "[8/8] Instalando custom nodes..."
 )
 
 # Aguardar SageAttention terminar
+log_info "Aguardando conclusão da instalação do SageAttention..."
 wait $SAGE_PID
 
 # Verificar instalação
@@ -378,7 +515,8 @@ ERROR_COUNT=0
 
 check_file() {
     if [ -f "$1" ]; then
-        log_success "  ✓ $(basename "$1")"
+        SIZE=$(du -h "$1" | cut -f1)
+        log_success "  ✓ $(basename "$1") ($SIZE)"
         return 0
     else
         log_error "  ✗ $(basename "$1") não encontrado"
@@ -388,6 +526,7 @@ check_file() {
 }
 
 # Verificar ComfyUI
+log_debug "Verificando ComfyUI..."
 check_file "$COMFY_DIR/main.py"
 
 # Verificar modelos principais
@@ -400,9 +539,15 @@ check_file "$MODELS_DIR/clip_vision/clip_vision_h.safetensors"
 check_file "$MODELS_DIR/controlnet/controlnet-union.safetensors"
 
 # Verificar se algum modelo Civitai foi baixado
-CIVITAI_COUNT=$(ls -1 "$MODELS_DIR/loras/"*.safetensors 2>/dev/null | wc -l)
+log_debug "Verificando modelos Civitai em $MODELS_DIR/loras..."
+CIVITAI_FILES=$(ls -1 "$MODELS_DIR/loras/"*.safetensors 2>/dev/null || true)
+CIVITAI_COUNT=$(echo "$CIVITAI_FILES" | grep -c safetensors || echo 0)
+
 if [ "$CIVITAI_COUNT" -gt 0 ]; then
-    log_success "  ✓ $CIVITAI_COUNT modelo(s) Civitai encontrado(s)"
+    log_success "  ✓ $CIVITAI_COUNT modelo(s) Civitai encontrado(s):"
+    echo "$CIVITAI_FILES" | while read file; do
+        [ -n "$file" ] && log_debug "    - $(basename "$file") ($(du -h "$file" | cut -f1))"
+    done
 else
     log_warn "  ⚠ Nenhum modelo Civitai encontrado (não crítico)"
 fi
@@ -418,11 +563,19 @@ if [ $ERROR_COUNT -eq 0 ]; then
     SAGE_FLAG=""
     if python3 -c "import sageattention" 2>/dev/null; then
         SAGE_FLAG="--use-sage-attention"
-        log_success "SageAttention disponível (compilado com $MAX_JOBS jobs)"
+        log_success "SageAttention disponível (compilado com MAX_JOBS=$MAX_JOBS)"
     else
         log_info "SageAttention não disponível"
     fi
     
+    # Mostrar informações do sistema
+    log_debug "Configuração final:"
+    log_debug "  CPU: $CPU_CORES cores"
+    if command -v nvidia-smi >/dev/null 2>&1; then
+        log_debug "  GPU: $(nvidia-smi --query-gpu=name --format=csv,noheader | head -n1)"
+        log_debug "  VRAM: $(nvidia-smi --query-gpu=memory.total --format=csv,noheader | head -n1)"
+    fi
+    
     echo ""
     log_info "Iniciando ComfyUI..."
     
@@ -436,6 +589,7 @@ if [ $ERROR_COUNT -eq 0 ]; then
     
     # Mudar para o diretório do ComfyUI e executar
     cd "$COMFY_DIR"
+    log_cmd "comfy launch -- $SAGE_FLAG --listen \"$COMFY_HOST\" --port \"$COMFY_PORT\""
     exec comfy launch -- $SAGE_FLAG --listen "$COMFY_HOST" --port "$COMFY_PORT"
 else
     log_error "Instalação completada com $ERROR_COUNT erro(s)"