Update setup_comfyui_wan22.sh

setup_comfyui_wan22.sh

@@ -1,6 +1,6 @@
 #!/usr/bin/env bash
-# setup_comfyui_wan22_optimized.sh
-# Script otimizado para instalação do ComfyUI com Wan 2.2
+# setup_comfyui_wan22_fixed.sh
+# Script corrigido para instalação do ComfyUI com Wan 2.2
 # Requisitos: Linux + Python3 + git + (opcional) NVCC/CUDA para SageAttention
 
 set -Eeuo pipefail
@@ -32,7 +32,7 @@ REPO_22="Comfy-Org/Wan_2.2_ComfyUI_Repackaged"
 REPO_21="Comfy-Org/Wan_2.1_ComfyUI_repackaged"
 REPO_CN="xinsir/controlnet-union-sdxl-1.0"
 
-# Token do Civitai (pode sobrescrever via env)
+# Token do Civitai
 CIVITAI_TOKEN="${CIVITAI_TOKEN:-4fcb2834969399006a736ee402b061e5}"
 
 # Configurações de performance
@@ -51,7 +51,6 @@ export NVCC_APPEND_FLAGS="--threads 8"
 # -----------------------------
 # Setup inicial
 # -----------------------------
-MODELS_DIR="$COMFY_DIR/models"
 TMPDL="$(mktemp -d)"
 cleanup() { 
     log_info "Limpando arquivos temporários..."
@@ -74,12 +73,6 @@ need git
 need wget
 need curl
 
-# Criar estrutura de diretórios
-log_info "Criando estrutura de diretórios..."
-mkdir -p "$MODELS_DIR"/{diffusion_models,loras,vae,text_encoders,clip_vision,controlnet}
-mkdir -p "$COMFY_DIR/custom_nodes"
-mkdir -p "$TMPDL"/{wan22,wan21,controlnet,civitai}
-
 # -----------------------------
 # Funções auxiliares
 # -----------------------------
@@ -100,54 +93,58 @@ clone_or_update() {
     fi
 }
 
-download_with_progress() {
-    local url="$1"
-    local output="$2"
-    local description="$3"
-    
-    log_info "Baixando: $description"
-    wget --progress=bar:force:noscroll \
-         --tries=3 \
-         --timeout=30 \
-         --continue \
-         -O "$output" \
-         "$url" 2>&1 | \
-         grep --line-buffered "%" | \
-         sed -u -e "s/^/    /"
-}
-
 # -----------------------------
 # Instalação principal
 # -----------------------------
 echo ""
 log_info "========================================="
-log_info "  ComfyUI + Wan 2.2 Setup Otimizado"
+log_info "  ComfyUI + Wan 2.2 Setup Corrigido"
 log_info "  CPU Cores: $CPU_CORES | Jobs: $MAX_JOBS"
 log_info "========================================="
 echo ""
 
-# [1/7] Atualizar pip e instalar ferramentas essenciais
-log_info "[1/7] Atualizando pip e instalando ferramentas..."
-python3 -m pip install --quiet --upgrade pip wheel setuptools uv >/dev/null 2>&1
+# [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
 python3 -m pip install --quiet --upgrade comfy-cli "huggingface_hub>=0.23.0" hf_transfer aria2 >/dev/null 2>&1
 log_success "Ferramentas instaladas"
 
-# [2/7] Instalar ComfyUI
-log_info "[2/7] Instalando ComfyUI (GPU NVIDIA, fast deps)..."
-if [ ! -d "$COMFY_DIR" ]; then
-    comfy --skip-prompt install --fast-deps --nvidia
+# [2/8] Instalar ComfyUI - SEMPRE PRIMEIRO!
+log_info "[2/8] Instalando ComfyUI (GPU NVIDIA, fast deps)..."
+
+# Verificação correta se ComfyUI já existe
+if [ -f "$COMFY_DIR/main.py" ] && [ -d "$COMFY_DIR/models" ]; then
+    log_warn "ComfyUI já existe em $COMFY_DIR, pulando instalação base"
 else
-    log_warn "ComfyUI já existe, pulando instalação base"
+    log_info "Instalando ComfyUI em $COMFY_DIR..."
+    # Remove diretório parcialmente criado se existir
+    [ -d "$COMFY_DIR" ] && rm -rf "$COMFY_DIR"
+    
+    # Instala ComfyUI
+    comfy --skip-prompt install --fast-deps --nvidia
+    
+    # Verifica se instalou corretamente
+    if [ ! -f "$COMFY_DIR/main.py" ]; then
+        log_error "Falha na instalação do ComfyUI!"
+        exit 1
+    fi
 fi
-log_success "ComfyUI pronto"
 
-# [3/7] Baixar modelos Wan 2.2 (em paralelo com aria2)
-log_info "[3/7] Iniciando downloads dos modelos Wan 2.2..."
+# Criar estrutura de diretórios APÓS garantir que ComfyUI existe
+MODELS_DIR="$COMFY_DIR/models"
+log_info "Criando estrutura de diretórios..."
+mkdir -p "$MODELS_DIR"/{diffusion_models,loras,vae,text_encoders,clip_vision,controlnet}
+mkdir -p "$COMFY_DIR/custom_nodes"
+mkdir -p "$TMPDL"/{wan22,wan21,controlnet}
+
+log_success "ComfyUI pronto e estrutura criada"
+
+# [3/8] Baixar modelos Wan 2.2
+log_info "[3/8] Iniciando downloads dos modelos Wan 2.2..."
 (
     set -Eeuo pipefail
     
-    # Wan 2.2
-    log_info "  → Baixando Wan 2.2 models..."
+    log_info "  → Baixando Wan 2.2 models (isso pode demorar)..."
     huggingface-cli download "$REPO_22" \
         split_files/diffusion_models/wan2.2_fun_control_low_noise_14B_fp8_scaled.safetensors \
         split_files/diffusion_models/wan2.2_fun_control_high_noise_14B_fp8_scaled.safetensors \
@@ -157,39 +154,46 @@ log_info "[3/7] Iniciando downloads dos modelos Wan 2.2..."
         split_files/text_encoders/umt5_xxl_fp16.safetensors \
         --local-dir "$TMPDL/wan22" \
         --local-dir-use-symlinks False \
-        --resume-download \
-        --quiet 2>/dev/null || true
-    
-    # Wan 2.1
-    log_info "  → Baixando Wan 2.1 clip vision..."
-    huggingface-cli download "$REPO_21" \
-        split_files/clip_vision/clip_vision_h.safetensors \
-        --local-dir "$TMPDL/wan21" \
-        --local-dir-use-symlinks False \
-        --resume-download \
-        --quiet 2>/dev/null || true
+        --resume-download
     
     # 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_success "  Modelos Wan 2.2 baixados"
+) &
+WAN22_PID=$!
+
+# [4/8] Baixar Wan 2.1 clip vision
+log_info "[4/8] Baixando Wan 2.1 clip vision..."
+(
+    set -Eeuo pipefail
+    
+    log_info "  → Baixando clip vision..."
+    huggingface-cli download "$REPO_21" \
+        split_files/clip_vision/clip_vision_h.safetensors \
+        --local-dir "$TMPDL/wan21" \
+        --local-dir-use-symlinks False \
+        --resume-download
+    
     cp -f "$TMPDL/wan21/split_files/clip_vision/"*.safetensors "$MODELS_DIR/clip_vision/" 2>/dev/null || true
     
-    log_success "  Modelos Wan baixados"
+    log_success "  Clip vision baixado"
 ) &
-WAN_PID=$!
+WAN21_PID=$!
 
-# [4/7] Baixar ControlNet (em paralelo)
-log_info "[4/7] Baixando ControlNet Union..."
+# [5/8] Baixar ControlNet
+log_info "[5/8] Baixando ControlNet Union..."
 (
     set -Eeuo pipefail
+    
     huggingface-cli download "$REPO_CN" \
         diffusion_pytorch_model_promax.safetensors \
         --local-dir "$TMPDL/controlnet" \
         --local-dir-use-symlinks False \
-        --resume-download \
-        --quiet 2>/dev/null || true
+        --resume-download
     
     cp -f "$TMPDL/controlnet/diffusion_pytorch_model_promax.safetensors" \
         "$MODELS_DIR/controlnet/controlnet-union.safetensors"
@@ -198,74 +202,80 @@ log_info "[4/7] Baixando ControlNet Union..."
 ) &
 CN_PID=$!
 
-# [5/7] Baixar modelo do Civitai (CORREÇÃO IMPORTANTE)
-log_info "[5/7] Baixando modelo do Civitai..."
+# [6/8] Baixar modelo do Civitai
+log_info "[6/8] Baixando modelo do Civitai..."
 (
     set -Eeuo pipefail
-    CIVITAI_URL="https://civitai.com/api/download/models/2122278"
-    CIVITAI_PARAMS="type=Model&format=SafeTensor&size=pruned&fp=fp16&token=${CIVITAI_TOKEN}"
-    FULL_URL="${CIVITAI_URL}?${CIVITAI_PARAMS}"
     
     cd "$MODELS_DIR/loras"
     
-    # Usar curl com retry e progress bar
-    curl -L \
-         --retry 3 \
-         --retry-delay 5 \
-         --connect-timeout 30 \
-         --max-time 1800 \
-         --progress-bar \
-         -H "User-Agent: Mozilla/5.0" \
-         -o "civitai_model_temp.safetensors" \
-         "$FULL_URL"
+    # URL correta sem aspas que quebram os parâmetros
+    CIVITAI_URL="https://civitai.com/api/download/models/2122278?type=Model&format=SafeTensor&size=pruned&fp=fp16&token=${CIVITAI_TOKEN}"
     
-    # Renomear com o nome correto do header Content-Disposition se possível
-    if [ -f "civitai_model_temp.safetensors" ]; then
-        # Tentar obter o nome real do arquivo
-        REAL_NAME=$(curl -sI -L "$FULL_URL" | grep -i content-disposition | sed -n 's/.*filename="\?\([^"]*\)"\?.*/\1/p' | head -1)
-        if [ -n "$REAL_NAME" ]; then
-            mv "civitai_model_temp.safetensors" "$REAL_NAME"
-            log_success "  Civitai modelo salvo como: $REAL_NAME"
-        else
-            mv "civitai_model_temp.safetensors" "civitai_2122278.safetensors"
-            log_success "  Civitai modelo salvo como: civitai_2122278.safetensors"
-        fi
-    fi
+    # Usar wget com content-disposition para pegar o nome correto
+    wget --content-disposition \
+         --tries=3 \
+         --timeout=60 \
+         --continue \
+         --progress=bar:force:noscroll \
+         "$CIVITAI_URL" 2>&1 | grep --line-buffered "%" | sed -u -e "s/^/    /" || {
+        log_warn "  Tentando com curl como fallback..."
+        curl -L \
+             --retry 3 \
+             --connect-timeout 30 \
+             --progress-bar \
+             -o "flux_unleashed_v2.safetensors" \
+             "$CIVITAI_URL"
+    }
+    
+    log_success "  Modelo Civitai baixado"
 ) &
 CIVITAI_PID=$!
 
-# [6/7] Instalar custom nodes (em paralelo)
-log_info "[6/7] Instalando custom nodes..."
+# [7/8] Aguardar downloads principais antes de instalar custom nodes
+log_info "[7/8] Aguardando downloads principais..."
+wait $WAN22_PID $WAN21_PID $CN_PID $CIVITAI_PID
+
+# [8/8] Agora sim instalar custom nodes (ComfyUI já existe com certeza)
+log_info "[8/8] Instalando custom nodes e SageAttention..."
+
+# Custom nodes
 (
     set -Eeuo pipefail
     CN_DIR="$COMFY_DIR/custom_nodes"
     
+    # Verificar se o diretório existe
+    if [ ! -d "$CN_DIR" ]; then
+        log_error "Diretório custom_nodes não existe! Criando..."
+        mkdir -p "$CN_DIR"
+    fi
+    
     declare -A REPOS=(
-        ["https://github.com/kijai/ComfyUI-Florence2"]="$CN_DIR/ComfyUI-Florence2"
-        ["https://github.com/kijai/ComfyUI-WanVideoWrapper"]="$CN_DIR/ComfyUI-WanVideoWrapper"
-        ["https://github.com/Fannovel16/ComfyUI-Frame-Interpolation"]="$CN_DIR/ComfyUI-Frame-Interpolation"
-        ["https://github.com/kijai/ComfyUI-GIMM-VFI"]="$CN_DIR/ComfyUI-GIMM-VFI"
-        ["https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite"]="$CN_DIR/ComfyUI-VideoHelperSuite"
-        ["https://github.com/kijai/ComfyUI-KJNodes"]="$CN_DIR/ComfyUI-KJNodes"
-        ["https://github.com/Fannovel16/comfyui_controlnet_aux"]="$CN_DIR/comfyui_controlnet_aux"
-        ["https://github.com/Artificial-Sweetener/comfyui-WhiteRabbit"]="$CN_DIR/comfyui-WhiteRabbit"
-        ["https://github.com/LucipherDev/ComfyUI-AniDoc"]="$CN_DIR/ComfyUI-AniDoc"
-        ["https://github.com/shiimizu/ComfyUI_smZNodes"]="$CN_DIR/ComfyUI_smZNodes"
-        ["https://github.com/yuvraj108c/ComfyUI-Dwpose-Tensorrt"]="$CN_DIR/ComfyUI-Dwpose-Tensorrt"
-        ["https://github.com/grmchn/ComfyUI-ProportionChanger"]="$CN_DIR/ComfyUI-ProportionChanger"
+        ["https://github.com/kijai/ComfyUI-Florence2"]="ComfyUI-Florence2"
+        ["https://github.com/kijai/ComfyUI-WanVideoWrapper"]="ComfyUI-WanVideoWrapper"
+        ["https://github.com/Fannovel16/ComfyUI-Frame-Interpolation"]="ComfyUI-Frame-Interpolation"
+        ["https://github.com/kijai/ComfyUI-GIMM-VFI"]="ComfyUI-GIMM-VFI"
+        ["https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite"]="ComfyUI-VideoHelperSuite"
+        ["https://github.com/kijai/ComfyUI-KJNodes"]="ComfyUI-KJNodes"
+        ["https://github.com/Fannovel16/comfyui_controlnet_aux"]="comfyui_controlnet_aux"
+        ["https://github.com/Artificial-Sweetener/comfyui-WhiteRabbit"]="comfyui-WhiteRabbit"
+        ["https://github.com/LucipherDev/ComfyUI-AniDoc"]="ComfyUI-AniDoc"
+        ["https://github.com/shiimizu/ComfyUI_smZNodes"]="ComfyUI_smZNodes"
+        ["https://github.com/yuvraj108c/ComfyUI-Dwpose-Tensorrt"]="ComfyUI-Dwpose-Tensorrt"
+        ["https://github.com/grmchn/ComfyUI-ProportionChanger"]="ComfyUI-ProportionChanger"
     )
     
     for url in "${!REPOS[@]}"; do
-        clone_or_update "$url" "${REPOS[$url]}" &
+        node_name="${REPOS[$url]}"
+        node_path="$CN_DIR/$node_name"
+        clone_or_update "$url" "$node_path"
     done
-    wait
     
     log_success "  Custom nodes instalados"
 ) &
 NODES_PID=$!
 
-# [7/7] Instalar SageAttention
-log_info "[7/7] Instalando SageAttention..."
+# SageAttention
 (
     set -Eeuo pipefail
     
@@ -275,43 +285,31 @@ log_info "[7/7] Instalando SageAttention..."
         
         python3 -m pip install --quiet "triton>=3.0.0" 2>/dev/null || true
         
-        if [ -d "SageAttention" ]; then
-            (cd SageAttention && git pull --ff-only --quiet)
-        else
-            git clone --depth 1 https://github.com/thu-ml/SageAttention.git --quiet
+        # Clonar no diretório temporário primeiro
+        SAGE_DIR="$TMPDL/SageAttention"
+        if [ ! -d "$SAGE_DIR" ]; then
+            git clone --depth 1 https://github.com/thu-ml/SageAttention.git "$SAGE_DIR" --quiet
         fi
         
-        cd SageAttention
-        python3 setup.py install --quiet 2>/dev/null || {
-            log_warn "  SageAttention falhou, continuando sem ele"
+        cd "$SAGE_DIR"
+        python3 setup.py install 2>/dev/null || {
+            log_warn "  SageAttention falhou na compilação, continuando sem ele"
         }
         
-        log_success "  SageAttention instalado"
+        # Testar se instalou
+        if python3 -c "import sageattention" 2>/dev/null; then
+            log_success "  SageAttention instalado com sucesso"
+        else
+            log_warn "  SageAttention não foi instalado corretamente"
+        fi
     else
         log_warn "  CUDA não detectado, pulando SageAttention"
     fi
 ) &
 SAGE_PID=$!
 
-# Aguardar todos os processos em paralelo
-log_info "Aguardando conclusão dos downloads e instalações..."
-echo -n "  Progresso: "
-
-PIDS=($WAN_PID $CN_PID $CIVITAI_PID $NODES_PID $SAGE_PID)
-TOTAL=${#PIDS[@]}
-COMPLETED=0
-
-while [ $COMPLETED -lt $TOTAL ]; do
-    COMPLETED=0
-    for pid in "${PIDS[@]}"; do
-        if ! kill -0 $pid 2>/dev/null; then
-            ((COMPLETED++))
-        fi
-    done
-    echo -ne "\r  Progresso: [$COMPLETED/$TOTAL]"
-    sleep 1
-done
-echo ""
+# Aguardar custom nodes e SageAttention
+wait $NODES_PID $SAGE_PID
 
 # Verificar instalação
 log_info "Verificando instalação..."
@@ -328,7 +326,11 @@ check_file() {
     fi
 }
 
+# Verificar ComfyUI
+check_file "$COMFY_DIR/main.py"
+
 # Verificar modelos principais
+log_info "Verificando modelos baixados..."
 check_file "$MODELS_DIR/diffusion_models/wan2.2_fun_control_low_noise_14B_fp8_scaled.safetensors"
 check_file "$MODELS_DIR/diffusion_models/wan2.2_fun_control_high_noise_14B_fp8_scaled.safetensors"
 check_file "$MODELS_DIR/vae/wan_2.1_vae.safetensors"
@@ -336,6 +338,14 @@ check_file "$MODELS_DIR/text_encoders/umt5_xxl_fp16.safetensors"
 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)
+if [ "$CIVITAI_COUNT" -gt 0 ]; then
+    log_success "  ✓ $CIVITAI_COUNT modelo(s) Civitai encontrado(s)"
+else
+    log_warn "  ⚠ Nenhum modelo Civitai encontrado (não crítico)"
+fi
+
 # Relatório final
 echo ""
 log_info "========================================="
@@ -347,13 +357,23 @@ if [ $ERROR_COUNT -eq 0 ]; then
     if python3 -c "import sageattention" 2>/dev/null; then
         SAGE_FLAG="--use-sage-attention"
         log_success "SageAttention disponível"
+    else
+        log_info "SageAttention não disponível (normal se não tiver CUDA)"
     fi
     
     echo ""
     log_info "Iniciando ComfyUI..."
-    log_info "URL: http://$(hostname -I | awk '{print $1}'):$COMFY_PORT"
+    
+    # Tentar obter IP local
+    LOCAL_IP=$(hostname -I 2>/dev/null | awk '{print $1}')
+    if [ -n "$LOCAL_IP" ]; then
+        log_info "URL Local: http://$LOCAL_IP:$COMFY_PORT"
+    fi
+    log_info "URL: http://localhost:$COMFY_PORT"
     echo ""
     
+    # Mudar para o diretório do ComfyUI antes de executar
+    cd "$COMFY_DIR"
     exec comfy launch -- $SAGE_FLAG --listen "$COMFY_HOST" --port "$COMFY_PORT"
 else
     log_error "Instalação completada com $ERROR_COUNT erro(s)"