Upload WANcomfyui_installer_fixed.sh

WANcomfyui_installer_fixed.sh

@@ -0,0 +1,662 @@
+#!/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
+# ============================================================================
+
+set -e
+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 ---
+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"
+
+# ============================================================================
+# FUNÇÕES DE UTILIDADE
+# ============================================================================
+
+log_info() {
+    local msg="[$(date '+%Y-%m-%d %H:%M:%S')] INFO: $1"
+    echo "$msg" | tee -a "$LOG_FILE" >&2
+}
+
+log_warning() {
+    local msg="[$(date '+%Y-%m-%d %H:%M:%S')] WARNING: $1"
+    echo "$msg" | tee -a "$LOG_FILE" >&2
+}
+
+log_error() {
+    local msg="[$(date '+%Y-%m-%d %H:%M:%S')] ERROR: $1"
+    echo "$msg" | tee -a "$LOG_FILE" >&2
+}
+
+command_exists() {
+    command -v "$1" >/dev/null 2>&1
+}
+
+# ============================================================================
+# VERIFICAÇÃO DE REQUISITOS
+# ============================================================================
+
+check_requirements() {
+    log_info "Verificando requisitos do sistema..."
+    
+    local missing_deps=()
+    
+    if ! command_exists python3; then
+        missing_deps+=("python3")
+    fi
+    
+    if ! command_exists git; then
+        missing_deps+=("git")
+    fi
+    
+    if ! command_exists gcc; then
+        missing_deps+=("gcc/build-essential")
+    fi
+    
+    if ! command_exists nvcc; then
+        log_warning "CUDA toolkit (nvcc) não encontrado. A compilação do SageAttention pode falhar."
+    fi
+    
+    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."
+        exit 1
+    fi
+    
+    # Verifica versão do Python
+    local python_version=$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
+    log_info "Python version: $python_version"
+    
+    if ! python3 -c 'import sys; exit(0 if sys.version_info >= (3,8) else 1)'; then
+        log_error "Python 3.8 ou superior é necessário"
+        exit 1
+    fi
+    
+    log_info "Todos os requisitos básicos estão presentes."
+}
+
+# ============================================================================
+# FUNÇÃO DE LIMPEZA
+# ============================================================================
+
+cleanup_existing_installation() {
+    log_info "Verificando instalações existentes em: $WORKSPACE_DIR"
+    
+    if [ -d "$WORKSPACE_DIR" ]; then
+        log_warning "Diretório existente encontrado: $WORKSPACE_DIR"
+        read -p "Deseja remover a instalação existente? (s/n): " -n 1 -r
+        echo
+        if [[ $REPLY =~ ^[Ss]$ ]]; then
+            log_info "Removendo a instalação existente..."
+            rm -rf "$WORKSPACE_DIR"
+            log_info "Diretório removido."
+        else
+            log_error "Instalação cancelada pelo usuário."
+            exit 1
+        fi
+    else
+        log_info "Nenhuma instalação anterior encontrada no local."
+    fi
+}
+
+# ============================================================================
+# INSTALAÇÃO DO COMFY-CLI
+# ============================================================================
+
+install_comfy_cli() {
+    log_info "Instalando/atualizando comfy-cli..."
+    
+    python3 -m pip install --upgrade pip
+    python3 -m pip install --upgrade comfy-cli || {
+        log_error "Falha ao instalar comfy-cli"
+        return 1
+    }
+    
+    # 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."
+    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\""
+        return 1
+    fi
+    
+    log_info "comfy-cli instalado: $(comfy --version 2>/dev/null || echo 'versão desconhecida')"
+    return 0
+}
+
+# ============================================================================
+# INSTALAÇÃO DO COMFYUI
+# ============================================================================
+
+install_comfyui() {
+    log_info "Instalando ComfyUI em: $WORKSPACE_DIR"
+    
+    # Instala ComfyUI com suporte NVIDIA
+    if ! comfy --workspace "$WORKSPACE_DIR" --skip-prompt install --nvidia; then
+        log_error "Falha ao instalar ComfyUI"
+        return 1
+    fi
+    
+    log_info "ComfyUI instalado com sucesso."
+    return 0
+}
+
+# ============================================================================
+# INSTALAÇÃO DO HUGGINGFACE HUB E HF_TRANSFER
+# ============================================================================
+
+install_hf_tools() {
+    log_info "Instalando HuggingFace Hub CLI e hf_transfer..."
+    
+    python3 -m pip install -U "huggingface_hub[cli]" hf_transfer || {
+        log_error "Falha ao instalar ferramentas HuggingFace"
+        return 1
+    }
+    
+    # Habilita o hf_transfer para downloads mais rápidos
+    export HF_HUB_ENABLE_HF_TRANSFER=1
+    
+    log_info "HuggingFace tools instalados com sucesso."
+    return 0
+}
+
+# ============================================================================
+# INSTALAÇÃO DO SAGEATTENTION 2.2.0 (COMPILAÇÃO OTIMIZADA)
+# ============================================================================
+
+install_sageattention() {
+    log_info "Instalando SageAttention 2.2.0 com 2++ (compilação otimizada)..."
+    
+    local sage_dir="$INSTALL_BASE_DIR/SageAttention"
+    
+    # Remove instalação anterior se existir
+    if [ -d "$sage_dir" ]; then
+        log_info "Removendo instalação anterior do SageAttention..."
+        rm -rf "$sage_dir"
+    fi
+    
+    # Instala dependências necessárias
+    log_info "Instalando dependências do SageAttention (Triton e utilidades)..."
+    python3 -m pip install 'triton>=3.0.0,<3.1.0' packaging ninja || {
+        log_error "Falha ao instalar dependências do SageAttention"
+        return 1
+    }
+    
+    # Clona o repositório
+    log_info "Clonando repositório SageAttention..."
+    if ! git clone https://github.com/thu-ml/SageAttention.git "$sage_dir"; then
+        log_error "Falha ao clonar repositório SageAttention"
+        return 1
+    fi
+    
+    cd "$sage_dir"
+    
+    # Configura variáveis de ambiente para compilação otimizada
+    log_info "Configurando ambiente de compilação otimizada..."
+    export EXT_PARALLEL=4
+    export NVCC_APPEND_FLAGS="--threads 8"
+    export MAX_JOBS=32
+    export TORCH_CUDA_ARCH_LIST="8.9"  # Ada architecture
+    
+    # Detecta automaticamente outras arquiteturas CUDA se disponível
+    if command_exists nvidia-smi; then
+        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"
+            fi
+        fi
+    fi
+    
+    # Compila e instala
+    log_info "Compilando SageAttention (isso pode demorar 5-10 minutos)..."
+    if ! python3 setup.py install >> "$LOG_FILE" 2>&1; then
+        log_error "Falha ao compilar SageAttention. Verifique $LOG_FILE para detalhes."
+        cd "$INSTALL_BASE_DIR"
+        return 1
+    fi
+    
+    cd "$INSTALL_BASE_DIR"
+    log_info "SageAttention 2.2.0 instalado com sucesso!"
+    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
+# ============================================================================
+
+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
+    fi
+    
+    log_error "Falha ao baixar o arquivo do workflow."
+    return 1
+}
+
+# ============================================================================
+# DOWNLOAD DE MODELOS WAN 2.2
+# ============================================================================
+
+download_models() {
+    log_info "Iniciando download dos modelos Wan 2.2..."
+    
+    # Criação de diretórios
+    local dirs=(
+        "$WORKSPACE_DIR/models/diffusion_models"
+        "$WORKSPACE_DIR/models/loras"
+        "$WORKSPACE_DIR/models/text_encoders"
+        "$WORKSPACE_DIR/models/vae"
+        "$WORKSPACE_DIR/models/clip_vision"
+    )
+    
+    for dir in "${dirs[@]}"; do
+        mkdir -p "$dir"
+    done
+    
+    # Lista de modelos Wan 2.2
+    local models=(
+        # Diffusion Models
+        "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"
+        
+        # LoRAs
+        "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"
+        
+        # Text Encoders
+        "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"
+        
+        # VAE (mantido do original)
+        "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"
+        
+        # CLIP Vision (mantido do original)
+        "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 failed_downloads=()
+    
+    # Download sequencial com progresso
+    local total=${#models[@]}
+    local current=0
+    
+    for model_info in "${models[@]}"; do
+        IFS=':' read -r url dest_dir filename <<< "$model_info"
+        current=$((current + 1))
+        
+        log_info "[$current/$total] Baixando modelo: $filename"
+        
+        if ! download_file_with_retry "$url" "$dest_dir" "$filename"; then
+            failed_downloads+=("$filename")
+            log_warning "Falha ao baixar: $filename. Continuando com outros downloads..."
+        fi
+    done
+    
+    if [ ${#failed_downloads[@]} -gt 0 ]; then
+        log_warning "Os seguintes modelos falharam ao baixar:"
+        for model in "${failed_downloads[@]}"; do
+            log_warning "  - $model"
+        done
+        log_warning "Você pode tentar baixá-los manualmente mais tarde."
+    else
+        log_info "Todos os modelos foram baixados com sucesso!"
+    fi
+    
+    return 0
+}
+
+# ============================================================================
+# INSTALAÇÃO DE CUSTOM NODES E DEPENDÊNCIAS
+# ============================================================================
+
+install_custom_nodes_and_deps() {
+    log_info "Instalando custom nodes e dependências..."
+    
+    # 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
+    
+    # Lista de custom nodes
+    local custom_nodes=(
+        "https://github.com/LAOGOU-666/Comfyui-Memory_Cleanup"
+        "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes"
+        "https://github.com/kijai/ComfyUI-KJNodes"
+        "https://github.com/kijai/ComfyUI-WanVideoWrapper.git"
+    )
+    
+    local failed_nodes=()
+    
+    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)"
+        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
+}
+
+# ============================================================================
+# VERIFICAÇÃO FINAL DA INSTALAÇÃO
+# ============================================================================
+
+verify_installation() {
+    log_info "Verificando a instalação..."
+    
+    local issues=()
+    
+    # Verifica ComfyUI
+    if [ ! -d "$WORKSPACE_DIR" ]; then
+        issues+=("Diretório ComfyUI não encontrado")
+    fi
+    
+    # Verifica workflow
+    if [ ! -f "$WORKFLOW_FILE" ]; then
+        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 SageAttention
+    if ! python3 -c "import sageattention" 2>/dev/null; then
+        log_warning "SageAttention não está instalado (opcional)"
+    fi
+    
+    if [ ${#issues[@]} -gt 0 ]; then
+        log_error "Problemas encontrados na instalação:"
+        for issue in "${issues[@]}"; do
+            log_error "  - $issue"
+        done
+        return 1
+    fi
+    
+    log_info "Instalação verificada com sucesso!"
+    return 0
+}
+
+# ============================================================================
+# INICIAR O COMFYUI
+# ============================================================================
+
+launch_comfyui() {
+    log_info "================================================="
+    log_info "Instalação concluída com sucesso!"
+    log_info "================================================="
+    log_info ""
+    log_info "Iniciando ComfyUI..."
+    log_info "  Porta: 8818"
+    log_info "  Workspace: $WORKSPACE_DIR"
+    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 ""
+    log_info "Pressione Ctrl+C para parar o servidor"
+    log_info "================================================="
+    
+    # Verifica se o SageAttention está instalado
+    local sage_args=""
+    if python3 -c "import sageattention" 2>/dev/null; then
+        sage_args="--use-sage-attention"
+        log_info "SageAttention habilitado!"
+    fi
+    
+    # Inicia o ComfyUI
+    exec comfy --workspace "$WORKSPACE_DIR" launch -- \
+        --fast \
+        --listen 0.0.0.0 \
+        --port 8818 \
+        $sage_args \
+        --preview-method latent2rgb
+}
+
+# ============================================================================
+# FUNÇÃO PRINCIPAL
+# ============================================================================
+
+main() {
+    # Inicializa log
+    echo "=== Instalação ComfyUI iniciada em $(date) ===" > "$LOG_FILE"
+    
+    log_info "================================================="
+    log_info "Instalador ComfyUI com Wan 2.2"
+    log_info "================================================="
+    
+    # Etapa 1: Verificar requisitos
+    check_requirements || exit 1
+    
+    # Etapa 2: Limpar instalação existente
+    cleanup_existing_installation
+    
+    # Etapa 3: Instalar comfy-cli
+    install_comfy_cli || exit 1
+    
+    # Etapa 4: Instalar ferramentas HuggingFace
+    install_hf_tools || exit 1
+    
+    # Etapa 5: Instalar ComfyUI
+    install_comfyui || exit 1
+    
+    # Etapa 6: Baixar workflow
+    download_workflow || log_warning "Workflow não foi baixado"
+    
+    # Etapa 7: Instalar custom nodes
+    install_custom_nodes_and_deps
+    
+    # Etapa 8: PROCESSOS PARALELOS - SageAttention e Downloads de Modelos
+    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 "================================================="
+    
+    # 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"
+    
+    local both_done=false
+    local sage_done=false
+    local models_done=false
+    
+    while [ "$both_done" = false ]; do
+        # Verifica SageAttention
+        if [ "$sage_done" = false ]; then
+            if ! ps -p $sage_pid > /dev/null 2>&1; then
+                wait $sage_pid
+                local sage_exit=$?
+                sage_done=true
+                if [ $sage_exit -eq 0 ]; then
+                    log_info "✓ SageAttention instalado com sucesso!"
+                else
+                    log_warning "⚠ SageAttention pode não ter sido instalado corretamente"
+                fi
+            fi
+        fi
+        
+        # Verifica Downloads
+        if [ "$models_done" = false ]; then
+            if ! ps -p $models_pid > /dev/null 2>&1; then
+                wait $models_pid
+                local models_exit=$?
+                models_done=true
+                if [ $models_exit -eq 0 ]; then
+                    log_info "✓ Downloads dos modelos concluídos!"
+                else
+                    log_warning "⚠ Alguns modelos podem não ter sido baixados"
+                fi
+            fi
+        fi
+        
+        # Verifica se ambos terminaram
+        if [ "$sage_done" = true ] && [ "$models_done" = true ]; then
+            both_done=true
+        else
+            # Mostra status a cada 30 segundos
+            sleep 30
+            if [ "$sage_done" = false ] && ps -p $sage_pid > /dev/null 2>&1; then
+                log_info "... SageAttention ainda compilando..."
+            fi
+            if [ "$models_done" = false ] && ps -p $models_pid > /dev/null 2>&1; then
+                log_info "... Downloads ainda em progresso..."
+            fi
+        fi
+    done
+    
+    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"
+        exit 1
+    }
+    
+    # Etapa 10: Iniciar ComfyUI
+    launch_comfyui
+}
+
+# ============================================================================
+# TRATAMENTO DE SINAIS E EXECUÇÃO
+# ============================================================================
+
+cleanup_on_exit() {
+    log_info "Interrompendo instalação..."
+    # Mata processos filhos
+    jobs -p | xargs -r kill 2>/dev/null
+    exit 1
+}
+
+# Captura sinais de interrupção
+trap cleanup_on_exit INT TERM
+
+# Executa o script principal
+main "$@"