Update setup_comfyui_wan22.sh

setup_comfyui_wan22.sh

@@ -348,8 +348,23 @@ log_info "[5/8] Baixando ControlNet Union..."
         "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"
+    # O arquivo pode estar direto no diretório ou em subdiretório
+    if [ -f "$TMPDL/controlnet/diffusion_pytorch_model_promax.safetensors" ]; then
+        cp -fv "$TMPDL/controlnet/diffusion_pytorch_model_promax.safetensors" \
+            "$MODELS_DIR/controlnet/controlnet-union.safetensors"
+    elif [ -f "$TMPDL/xinsir/controlnet-union-sdxl-1.0/diffusion_pytorch_model_promax.safetensors" ]; then
+        cp -fv "$TMPDL/xinsir/controlnet-union-sdxl-1.0/diffusion_pytorch_model_promax.safetensors" \
+            "$MODELS_DIR/controlnet/controlnet-union.safetensors"
+    else
+        # Procurar o arquivo em qualquer lugar do diretório temporário
+        FOUND_FILE=$(find "$TMPDL" -name "diffusion_pytorch_model_promax.safetensors" -type f | head -n1)
+        if [ -n "$FOUND_FILE" ]; then
+            log_debug "ControlNet encontrado em: $FOUND_FILE"
+            cp -fv "$FOUND_FILE" "$MODELS_DIR/controlnet/controlnet-union.safetensors"
+        else
+            log_warn "ControlNet não encontrado no diretório temporário"
+        fi
+    fi
     
     log_success "  ControlNet baixado"
 ) &
@@ -366,43 +381,37 @@ log_info "[6/8] Baixando modelo do Civitai..."
     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 \
-              "$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 "$FILENAME" \
-             "$CIVITAI_URL"
+    
+    # Usar curl direto pois wget está falhando com nomes longos
+    # Primeiro obter o nome real do arquivo
+    log_debug "Obtendo nome do arquivo..."
+    HEADERS=$(curl -sI -L --max-time 30 "$CIVITAI_URL")
+    FILENAME=$(echo "$HEADERS" | grep -i "content-disposition" | sed -n 's/.*filename="\([^"]*\)".*/\1/p' | tr -d '\r')
+    
+    if [ -z "$FILENAME" ]; then
+        # Tentar outro padrão de header
+        FILENAME=$(echo "$HEADERS" | grep -i "content-disposition" | sed -n "s/.*filename='\([^']*\)'.*/\1/p" | tr -d '\r')
     fi
     
-    # 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"
+    if [ -z "$FILENAME" ]; then
+        # Se ainda não conseguiu, usar nome padrão baseado no ID do modelo
+        FILENAME="civitai_model_2122278.safetensors"
+        log_warn "Não foi possível detectar nome do arquivo, usando: $FILENAME"
+    else
+        log_success "Nome do arquivo detectado: $FILENAME"
+    fi
+    
+    # Baixar o arquivo
+    log_cmd "curl -L --progress-bar \"$CIVITAI_URL\" -o \"$FILENAME\""
+    
+    if curl -L \
+         --retry 3 \
+         --connect-timeout 60 \
+         --max-time 1800 \
+         --progress-bar \
+         -o "$FILENAME" \
+         "$CIVITAI_URL"; then
+        log_success "  Modelo Civitai baixado: $FILENAME"
     else
         log_error "  Falha ao baixar modelo Civitai"
     fi
@@ -423,9 +432,9 @@ log_info "[7/8] Preparando SageAttention (paralelo com downloads)..."
         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
+        # Instalar pynvml para melhor detecção de GPU (ignorar warning de deprecation)
+        log_cmd "python3 -m pip install nvidia-ml-py"
+        python3 -m pip install nvidia-ml-py 2>&1 | grep -v "FutureWarning" || true
         
         # Clonar SageAttention
         SAGE_DIR="$TMPDL/SageAttention"
@@ -442,21 +451,9 @@ log_info "[7/8] Preparando SageAttention (paralelo com downloads)..."
         
         # Mostrar output completo da compilação
         log_cmd "python3 setup.py install"
-        python3 setup.py install
+        python3 setup.py install 2>&1 | grep -v "SetuptoolsDeprecationWarning" | grep -v "FutureWarning" || true
         
-        # 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
+        log_success "  SageAttention instalado"
     else
         log_warn "  CUDA não detectado, pulando SageAttention"
     fi
@@ -559,15 +556,9 @@ if [ $ERROR_COUNT -eq 0 ]; then
     log_success "Instalação concluída com sucesso!"
     log_info "ComfyUI instalado em: $COMFY_DIR"
     
-    # Verificar se SageAttention está disponível
-    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=$MAX_JOBS)"
-    else
-        SAGE_FLAG="--use-sage-attention"
-        log_info "SageAttention não disponível"
-    fi
+    # Sempre usar flag do SageAttention (ComfyUI ignora se não estiver instalado)
+    SAGE_FLAG="--use-sage-attention"
+    log_info "Flag SageAttention habilitada (será usada se disponível)"
     
     # Mostrar informações do sistema
     log_debug "Configuração final:"