Update setup_comfy_wanDANCE.sh

setup_comfy_wanDANCE.sh

@@ -43,13 +43,13 @@ export EXT_PARALLEL=4
 # Formato: "URL|TIPO|NOME_ARQUIVO_OPCIONAL"
 readonly DOWNLOAD_FILES=(
     # Wan 2.2 models
-    "hf://Kijai/WanVideo_comfy/resolve/main/Wan2_1-I2V-14B-720P_fp8_e4m3fn.safetensors|diffusion_models|Wan2_1-I2V-14B-720P_fp8_e4m3fn.safetensors"
-    "hf://Kijai/WanVideo_comfy/resolve/main/Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank256_bf16.safetensors|loras|lightx2v_I2V_14B_480p_cfg_step_distill_rank256_bf16.safetensors"
-    "hf://Comfy-Org/Wan_2.2_ComfyUI_Repackaged/split_files/vae/wan_2.1_vae.safetensors|vae|"
-    "hf://Comfy-Org/Wan_2.2_ComfyUI_Repackaged/split_files/text_encoders/umt5_xxl_fp16.safetensors|text_encoders|"
+    "hf://Kijai/WanVideo_comfy/Wan2_1-I2V-14B-720P_fp8_e4m3fn.safetensors|diffusion_models|Wan2_1-I2V-14B-720P_fp8_e4m3fn.safetensors"
+    "hf://Kijai/WanVideo_comfy/Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank256_bf16.safetensors|loras|lightx2v_I2V_14B_480p_cfg_step_distill_rank256_bf16.safetensors"
+    "hf://Comfy-Org/Wan_2.2_ComfyUI_Repackaged/split_files/vae/wan_2.1_vae.safetensors|vae|wan_2.1_vae.safetensors"
+    "hf://Comfy-Org/Wan_2.2_ComfyUI_Repackaged/split_files/text_encoders/umt5_xxl_fp16.safetensors|text_encoders|umt5_xxl_fp16.safetensors"
     # Wan 2.1 clip vision
-    "hf://Comfy-Org/Wan_2.1_ComfyUI_repackaged/split_files/clip_vision/clip_vision_h.safetensors|clip_vision|"
-    # ControlNet   models/vae_approx.
+    "hf://Comfy-Org/Wan_2.1_ComfyUI_repackaged/split_files/clip_vision/clip_vision_h.safetensors|clip_vision|clip_vision_h.safetensors"
+    # ControlNet e outros modelos
     "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/taew2_1.safetensors|vae_approx|taew2_1.safetensors"
     "https://huggingface.co/Kijai/WanVideo_comfy/resolve/main/UniAnimate-Wan2.1-14B-Lora-12000-fp16.safetensors|loras|UniAnimate-Wan2.1-14B-Lora-12000-fp16.safetensors"
     "hf://xinsir/controlnet-union-sdxl-1.0/diffusion_pytorch_model_promax.safetensors|controlnet|controlnet-union.safetensors"
@@ -92,14 +92,14 @@ add_civitai_token() {
     fi
 }
 
-# Download HuggingFace usando huggingface-cli com hf_transfer
+# Download HuggingFace usando hf com hf_transfer
 download_hf() {
     local repo="$1"
     local file_path="$2"
     local target_dir="$3"
     local filename="$4"
     
-    # Extrair nome do arquivo se não fornecido
+    # Se filename não foi fornecido, extrair do caminho
     if [ -z "$filename" ]; then
         filename=$(basename "$file_path")
     fi
@@ -113,28 +113,49 @@ download_hf() {
     fi
     
     log_info "Baixando de HF: $filename"
+    log_info "  Repo: $repo"
+    log_info "  Path: $file_path"
     
-    # Usar huggingface-cli com hf_transfer habilitado para velocidade máxima
-    if HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo" "$file_path" \
-        --local-dir "$target_dir" \
-        --local-dir-use-symlinks False \
-        --resume-download 2>/dev/null; then
-        
-        # Verificar se arquivo foi baixado
-        if [ -f "$target_file" ]; then
-            log_success "Download concluído: $filename"
-            return 0
+    # Criar diretório temporário para download
+    local temp_dir=$(mktemp -d)
+    
+    # Tentar primeiro com 'hf download' (novo comando)
+    local download_success=false
+    
+    if command_exists hf; then
+        if HF_HUB_ENABLE_HF_TRANSFER=1 hf download "$repo" "$file_path" \
+            --local-dir "$temp_dir" \
+            --local-dir-use-symlinks False 2>/dev/null; then
+            download_success=true
+        fi
+    fi
+    
+    # Se 'hf' não funcionou, tentar com huggingface-cli
+    if [ "$download_success" = false ] && command_exists huggingface-cli; then
+        log_warn "Tentando com huggingface-cli..."
+        if HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo" "$file_path" \
+            --local-dir "$temp_dir" \
+            --local-dir-use-symlinks False 2>/dev/null; then
+            download_success=true
         fi
+    fi
+    
+    if [ "$download_success" = true ]; then
+        # Procurar o arquivo baixado (pode estar em subdiretórios)
+        local downloaded_file=$(find "$temp_dir" -type f \( -name "*.safetensors" -o -name "*.pth" -o -name "*.pt" -o -name "*.bin" \) 2>/dev/null | head -1)
         
-        # Procurar arquivo em subdiretórios se não estiver no local esperado
-        local downloaded_file=$(find "$target_dir" -name "$filename" -type f 2>/dev/null | head -1)
-        if [ -n "$downloaded_file" ] && [ "$downloaded_file" != "$target_file" ]; then
+        if [ -n "$downloaded_file" ]; then
+            # Mover para o destino correto com o nome correto
             mv "$downloaded_file" "$target_file"
-            log_success "Download concluído e movido: $filename"
+            rm -rf "$temp_dir"
+            log_success "Download concluído: $filename"
             return 0
+        else
+            log_error "Arquivo baixado mas não encontrado em $temp_dir"
         fi
     fi
     
+    rm -rf "$temp_dir"
     log_error "Falha ao baixar: $filename"
     return 1
 }
@@ -209,6 +230,7 @@ process_downloads() {
         local target_dir="$MODELS_DIR"
         case "$type" in
             checkpoints) target_dir="$MODELS_DIR/checkpoints" ;;
+            diffusion_models) target_dir="$MODELS_DIR/diffusion_models" ;;
             loras) target_dir="$MODELS_DIR/loras" ;;
             vae) target_dir="$MODELS_DIR/vae" ;;
             vae_approx) target_dir="$MODELS_DIR/vae_approx" ;;
@@ -226,8 +248,10 @@ process_downloads() {
         if [[ "$url" == hf://* ]]; then
             # HuggingFace download
             url="${url#hf://}"  # Remove prefixo hf://
-            repo=$(echo "$url" | cut -d'/' -f1-2)
-            file_path=$(echo "$url" | cut -d'/' -f3-)
+            
+            # Separar repo e caminho do arquivo
+            local repo=$(echo "$url" | cut -d'/' -f1-2)
+            local file_path=$(echo "$url" | cut -d'/' -f3-)
             
             download_hf "$repo" "$file_path" "$target_dir" "$filename" || ((failed_count++))
         else
@@ -236,20 +260,6 @@ process_downloads() {
         fi
     done
     
-    # Limpar arquivos órfãos em split_files (se existirem)
-    if [ -d "$MODELS_DIR/diffusion_models/split_files" ]; then
-        log_info "Movendo arquivos de split_files para diretórios corretos..."
-        find "$MODELS_DIR" -path "*/split_files/*" -name "*.safetensors" | while read -r file; do
-            local base_dir=$(dirname "$(dirname "$file")")
-            local target_name=$(basename "$file")
-            if [ ! -f "$base_dir/$target_name" ]; then
-                mv "$file" "$base_dir/" 2>/dev/null && log_success "Movido: $target_name"
-            fi
-        done
-        # Remover diretórios split_files vazios
-        find "$MODELS_DIR" -type d -name "split_files" -empty -delete 2>/dev/null
-    fi
-    
     return $failed_count
 }
 
@@ -289,7 +299,7 @@ clone_or_update() {
 # -----------------------------
 echo ""
 log_info "========================================="
-log_info " ComfyUI + Wan 2.2 Setup (Fixed)"
+log_info " ComfyUI + Wan 2.2 Setup (Fixed v2)"
 log_info "========================================="
 echo ""
 
@@ -315,7 +325,7 @@ log_info "[2/8] Preparando ferramentas Python..."
 python3 -m pip install --upgrade pip wheel setuptools -q
 
 # Instalar huggingface_hub com CLI e hf_transfer para velocidade máxima
-python3 -m pip install --upgrade "huggingface_hub[cli]>=0.26.0" hf_transfer comfy-cli -q
+python3 -m pip install --upgrade "huggingface_hub[cli,hf_transfer]>=0.26.0" comfy-cli -q
 
 # Configurar token HF se disponível (opcional)
 HF_TOKEN="${HF_TOKEN:-}"
@@ -327,11 +337,13 @@ if [ -n "$HF_TOKEN" ]; then
     fi
 fi
 
-# Verificar se hf está disponível
-if command_exists huggingface-cli; then
-    log_success "huggingface-cli disponível"
+# Verificar comandos disponíveis
+if command_exists hf; then
+    log_success "Comando 'hf' disponível (recomendado)"
+elif command_exists huggingface-cli; then
+    log_warn "Usando 'huggingface-cli' (deprecated, mas funcional)"
 else
-    log_warn "huggingface-cli não encontrado"
+    log_error "Nenhum comando HF disponível!"
 fi
 
 log_success "Ferramentas instaladas"
@@ -374,7 +386,7 @@ fi
 
 # [6] Baixar modelos
 log_info "[6/8] Baixando modelos..."
-mkdir -p "$MODELS_DIR"/{diffusion_models,loras,vae,text_encoders,clip_vision,controlnet,upscale_models}
+mkdir -p "$MODELS_DIR"/{diffusion_models,loras,vae,vae_approx,text_encoders,clip_vision,controlnet,upscale_models,checkpoints}
 
 # Verificar ferramentas de download
 if command_exists aria2c; then
@@ -407,14 +419,14 @@ log_success "Custom nodes instalados"
 echo ""
 log_info "[8/8] Verificando instalação..."
 
-# Verificar arquivos principais
+# Verificar arquivos principais (ajustado para os modelos que você está baixando)
 check_critical_files() {
     local files=(
         "$COMFY_DIR/main.py"
-        "$MODELS_DIR/diffusion_models/wan2.2_fun_control_low_noise_14B_fp8_scaled.safetensors"
-        "$MODELS_DIR/diffusion_models/wan2.2_fun_control_high_noise_14B_fp8_scaled.safetensors"
+        "$MODELS_DIR/diffusion_models/Wan2_1-I2V-14B-720P_fp8_e4m3fn.safetensors"
         "$MODELS_DIR/vae/wan_2.1_vae.safetensors"
         "$MODELS_DIR/text_encoders/umt5_xxl_fp16.safetensors"
+        "$MODELS_DIR/clip_vision/clip_vision_h.safetensors"
     )
     
     local missing=0
@@ -422,12 +434,23 @@ check_critical_files() {
         if [ -f "$file" ]; then
             log_success "✓ $(basename "$file")"
         else
-            log_error "✗ $(basename "$file") não encontrado"
-            ((missing++))
+            log_warn "✗ $(basename "$file") não encontrado (opcional)"
+        fi
+    done
+    
+    # Listar modelos baixados
+    echo ""
+    log_info "Modelos disponíveis:"
+    for dir in diffusion_models loras vae vae_approx text_encoders clip_vision controlnet upscale_models checkpoints; do
+        if [ -d "$MODELS_DIR/$dir" ]; then
+            local count=$(find "$MODELS_DIR/$dir" -type f \( -name "*.safetensors" -o -name "*.pth" -o -name "*.pt" \) 2>/dev/null | wc -l)
+            if [ $count -gt 0 ]; then
+                log_success "  $dir: $count arquivo(s)"
+            fi
         fi
     done
     
-    return $missing
+    return 0
 }
 
 if check_critical_files; then
@@ -448,11 +471,9 @@ if check_critical_files; then
     cd "$COMFY_DIR"
     exec comfy launch -- $SAGE_FLAG --listen "$COMFY_HOST" --port "$COMFY_PORT"
 else
-    log_error "Arquivos críticos faltando. Verifique os erros acima."
-    
-    # Mostrar onde os arquivos podem estar
-    log_info "Procurando arquivos mal posicionados..."
-    find "$MODELS_DIR" -name "*.safetensors" -type f | head -20
+    log_error "Alguns arquivos não foram encontrados, mas a instalação pode funcionar."
+    log_info "Iniciando ComfyUI mesmo assim..."
     
-    exit 1
+    cd "$COMFY_DIR"
+    exec comfy launch -- --listen "$COMFY_HOST" --port "$COMFY_PORT"
 fi