Upload wanForProduction.sh

wanForProduction.sh

@@ -0,0 +1,363 @@
+#!/usr/bin/env bash
+# setup_comfyui_combined.sh
+# Script combinado para instalação do ComfyUI com Wan 2.2
+
+set -euo pipefail
+
+# Cores
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+NC='\033[0m'
+
+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"; }
+
+# Configuração
+COMFY_DIR="/root/comfy/ComfyUI"
+MODELS_DIR="$COMFY_DIR/models"
+COMFY_HOST="0.0.0.0"
+COMFY_PORT="8818"
+CIVITAI_TOKEN="4fcb2834969399006a736ee402b061e5"
+
+# Performance
+export MAX_JOBS=32
+export NVCC_APPEND_FLAGS="--threads 8"
+export UV_SYSTEM_PYTHON=1
+export PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"
+export HF_HUB_ENABLE_HF_TRANSFER=1
+export HF_TRANSFER_CONCURRENCY=16
+export EXT_PARALLEL=4
+
+# Lista de downloads - Formato: "URL|TIPO|NOME_OPCIONAL"
+DOWNLOAD_FILES=(
+    # Wan 2.2 models
+    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_HIGH_fp16.safetensors|loras|Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_HIGH_fp16.safetensors"
+    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_LOW_fp16.safetensors|loras|Wan2.2-Lightning_T2V-v1.1-A14B-4steps-lora_LOW_fp16.safetensors"
+    "hf://Kijai/WanVideo_comfy_fp8_scaled/I2V/Wan2_2-I2V-A14B-HIGH_fp8_e5m2_scaled_KJ.safetensors|diffusion_models|Wan2_2-I2V-A14B-HIGH_fp8_e5m2_scaled_KJ.safetensors"
+    "hf://Kijai/WanVideo_comfy_fp8_scaled/I2V/Wan2_2-I2V-A14B-LOW_fp8_e5m2_scaled_KJ.safetensors|diffusion_models|Wan2_2-I2V-A14B-LOW_fp8_e5m2_scaled_KJ.safetensors"
+    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_I2V-A14B-4steps-lora_HIGH_fp16.safetensors|loras|Wan2.2-Lightning_I2V-A14B-4steps-lora_HIGH_fp16.safetensors"
+    "hf://Kijai/WanVideo_comfy/Wan22-Lightning/Wan2.2-Lightning_I2V-A14B-4steps-lora_LOW_fp16.safetensors|loras|Wan2.2-Lightning_I2V-A14B-4steps-lora_LOW_fp16.safetensors"
+    "hf://Kijai/WanVideo_comfy/Wan2_1_VAE_fp32.safetensors|vae|Wan2_1_VAE_fp32.safetensors"
+    "hf://Kijai/WanVideo_comfy/umt5-xxl-enc-bf16.safetensors|text_encoders|umt5-xxl-enc-bf16.safetensors"
+    
+    # Wan 2.1 models
+    "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://Kijai/WanVideo_comfy/open-clip-xlm-roberta-large-vit-huge-14_visual_fp32.safetensors|clip_vision|open-clip-xlm-roberta-large-vit-huge-14_visual_fp32.safetensors"
+    "hf://Comfy-Org/Wan_2.1_ComfyUI_repackaged/split_files/clip_vision/clip_vision_h.safetensors|clip_vision|clip_vision_h.safetensors"
+    
+    # Upscaler extra
+    "https://huggingface.co/ABDALLALSWAITI/Upscalers/resolve/main/anime/2x-AnimeSharpV2_MoSR_Soft.pth|upscale_models|2x-AnimeSharpV2_MoSR_Soft.pth"
+)
+
+# Mega downloads
+MEGA_LORAS=(
+    "https://mega.nz/file/QFxREJpZ#0dge8TyRgb52lMFKdsk-Ac-rwcoG49hvBHvaqNsZkuk"
+    "https://mega.nz/file/lIhQETjT#nKlQ01Euw6fUxo8P5bfwlQrxAkWEedXQQOqHv5u4yrM"
+    "https://mega.nz/file/UIAESACY#Qlei1Pj5Nwno3Sz-wsTekvD-YgCkN1A5QbFKoRHXG9E"
+    "https://mega.nz/file/cdAg1Q7D#tQVrer2sB-M7XdYiKAqr7kRRfJw4pXCdliOg2aPJUGE"
+    "https://mega.nz/file/5RQFiAAJ#QCWrHkuJjXlFufR18g5R9YXJGqJ1pQp7xbqNRgSYUjc"
+    "https://mega.nz/file/tABgxLTS#NNlPV8NPc7R52MIjABMFUb41EOFDxFoszBxOrPMN13Q"
+    "https://mega.nz/file/4MxBgJ5Y#R8OSby0u-t-QFBnpnOtdjN3xuiVRvHa4_OEzTwWoGKQ"
+    "https://mega.nz/file/dJJxiSRR#o3XsYfSvb04xg7gNVdfHb3XTVHgvx0ZyXe9N8VQ9Lfw"
+    "https://mega.nz/file/wcAFhRJZ#ifZcckts0wPq1FG693tYLkpFrvjXvab1ZQeL6DuL1T8"
+    "https://mega.nz/file/wNIgmRwb#vn6rKM3QuTDpWVTLEBDsKjIkNkaX2wLKeauKNeVkcFs"
+    "https://mega.nz/file/hZhhhCQS#4je0Im2g1kEvTCuPotjVWkA-g6XICkqMb-mKxMt6R3Y"
+)
+
+MEGA_UPSCALE=(
+    "https://mega.nz/file/qdQwAJSC#rPY2Hg4x-_RbHiHUBFW-YU8xZUqPpq5Gpl1uPFiJvW4"
+    "https://mega.nz/file/KIh2marA#CehVNXxQhGfp1bqUL0K1YPWNqrCUoLPCZGq2W6XqmWo"
+)
+
+# Custom nodes do CODE2
+CUSTOM_NODES=(
+    "https://github.com/kijai/ComfyUI-Florence2"
+    "https://github.com/adbrasi/groqrouter"
+    "https://github.com/kijai/ComfyUI-WanVideoWrapper"
+    "https://github.com/Fannovel16/ComfyUI-Frame-Interpolation"
+    "https://github.com/kijai/ComfyUI-GIMM-VFI"
+    "https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite"
+    "https://github.com/kijai/ComfyUI-KJNodes"
+    "https://github.com/Fannovel16/comfyui_controlnet_aux"
+    "https://github.com/Artificial-Sweetener/comfyui-WhiteRabbit"
+    "https://github.com/shiimizu/ComfyUI_smZNodes"
+    "https://github.com/CoreyCorza/ComfyUI-CRZnodes"
+    "https://github.com/pythongosssss/ComfyUI-Custom-Scripts"
+    "https://github.com/pythongosssss/ComfyUI-WD14-Tagger"
+    "https://github.com/WASasquatch/was-node-suite-comfyui"
+    "https://github.com/justUmen/Bjornulf_custom_nodes"
+    "https://github.com/mingsky-ai/ComfyUI-MingNodes"
+    "https://github.com/numz/ComfyUI-SeedVR2_VideoUpscaler"
+    "https://github.com/1038lab/ComfyUI-RMBG"
+    "https://github.com/wallish77/wlsh_nodes"
+    "https://github.com/yuvraj108c/ComfyUI-Upscaler-Tensorrt"
+    "https://github.com/raindrop313/ComfyUI-WanVideoStartEndFrames"
+    "https://github.com/MushroomFleet/DJZ-Nodes"
+    "https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes"
+    "https://github.com/grmchn/ComfyUI-ProportionChanger"
+)
+
+# Download HuggingFace
+download_hf() {
+    local url="$1"
+    local target_dir="$2"
+    local filename="$3"
+    
+    # Extrair repo e file_path da URL hf://
+    url="${url#hf://}"
+    local repo="$(echo "$url" | cut -d'/' -f1-2)"
+    local file_path="$(echo "$url" | cut -d'/' -f3-)"
+    
+    if [ -n "$filename" ] && [ -f "$target_dir/$filename" ]; then
+        log_success "Já existe: $filename"
+        return 0
+    fi
+    
+    log_info "Baixando HF: $filename"
+    
+    local temp_dir
+    temp_dir="$(mktemp -d)"
+    
+    if command -v hf >/dev/null 2>&1; then
+        HF_HUB_ENABLE_HF_TRANSFER=1 hf download "$repo" "$file_path" \
+            --local-dir "$temp_dir" --local-dir-use-symlinks False 2>/dev/null || {
+            rm -rf "$temp_dir"
+            log_error "Falha ao baixar: $filename"
+            return 1
+        }
+    elif command -v huggingface-cli >/dev/null 2>&1; then
+        HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download "$repo" "$file_path" \
+            --local-dir "$temp_dir" --local-dir-use-symlinks False 2>/dev/null || {
+            rm -rf "$temp_dir"
+            log_error "Falha ao baixar: $filename"
+            return 1
+        }
+    else
+        wget -q --show-progress -c -O "$target_dir/$filename" "https://huggingface.co/$url"
+        rm -rf "$temp_dir"
+        return $?
+    fi
+    
+    # Mover arquivo baixado para destino
+    local downloaded_file
+    downloaded_file="$(find "$temp_dir" -type f \( -name "*.safetensors" -o -name "*.pth" -o -name "*.pt" -o -name "*.bin" \) | head -1)"
+    if [ -n "$downloaded_file" ]; then
+        mv "$downloaded_file" "$target_dir/$filename"
+        rm -rf "$temp_dir"
+        log_success "Download concluído: $filename"
+        return 0
+    else
+        rm -rf "$temp_dir"
+        log_error "Arquivo não encontrado após download: $filename"
+        return 1
+    fi
+}
+
+# Download Mega
+download_mega() {
+    local url="$1"
+    local target_dir="$2"
+    
+    log_info "Baixando do Mega..."
+    
+    cd "$target_dir"
+    if megadl "$url" 2>/dev/null; then
+        log_success "Mega download OK"
+    else
+        log_warn "Mega download falhou"
+    fi
+    cd - >/dev/null
+}
+
+# Download normal
+download_file() {
+    local url="$1"
+    local target_dir="$2"
+    local filename="$3"
+    
+    # Adicionar token Civitai se necessário
+    if [[ "$url" == *"civitai.com"* ]] && [[ "$url" != *"token="* ]]; then
+        url="${url}&token=${CIVITAI_TOKEN}"
+    fi
+    
+    if [ -n "$filename" ] && [ -f "$target_dir/$filename" ]; then
+        log_success "Já existe: $filename"
+        return 0
+    fi
+    
+    log_info "Baixando: ${filename:-arquivo}"
+    
+    # Tentar com aria2c primeiro
+    if command -v aria2c >/dev/null 2>&1; then
+        if [ -n "$filename" ]; then
+            aria2c -c -x 16 -s 16 -k 1M --console-log-level=warn --dir="$target_dir" --out="$filename" "$url" || \
+            wget -q --show-progress -c -O "$target_dir/$filename" "$url" || \
+            log_error "Download falhou: $filename"
+        else
+            aria2c -c -x 16 -s 16 -k 1M --console-log-level=warn --dir="$target_dir" "$url" || \
+            (cd "$target_dir" && wget -q --show-progress -c "$url") || \
+            log_error "Download falhou: $url"
+        fi
+    elif [ -n "$filename" ]; then
+        wget -q --show-progress -c -O "$target_dir/$filename" "$url" || \
+        log_error "Download falhou: $filename"
+    else
+        (cd "$target_dir" && wget -q --show-progress -c "$url") || \
+        log_error "Download falhou: $url"
+    fi
+}
+
+# Processar downloads
+process_downloads() {
+    for entry in "${DOWNLOAD_FILES[@]}"; do
+        IFS='|' read -r url type filename <<< "$entry"
+        
+        # Criar diretório
+        local target_dir="$MODELS_DIR/$type"
+        mkdir -p "$target_dir"
+        
+        # Baixar arquivo
+        if [[ "$url" == hf://* ]]; then
+            download_hf "$url" "$target_dir" "$filename"
+        else
+            download_file "$url" "$target_dir" "$filename"
+        fi
+    done
+}
+
+# Clone git repo
+clone_repo() {
+    local url="$1"
+    local dest="$2"
+    
+    if [ -d "$dest/.git" ]; then
+        git -C "$dest" pull --ff-only 2>/dev/null || true
+    else
+        git clone --depth 1 "$url" "$dest" 2>/dev/null || true
+    fi
+    
+    # Instalar requirements se existir
+    if [ -f "$dest/requirements.txt" ]; then
+        pip install -q -r "$dest/requirements.txt" 2>/dev/null || true
+    fi
+}
+
+# ========== INSTALAÇÃO ==========
+
+log_info "========================================="
+log_info " ComfyUI + Wan 2.2 Setup"
+log_info "========================================="
+
+# 1. Instalar dependências
+log_info "[1/7] Instalando dependências..."
+
+apt-get update -qq
+apt-get install -y -qq aria2 megatools git wget curl 2>/dev/null
+
+pip install --upgrade pip wheel setuptools -q
+pip install --upgrade "huggingface_hub[cli,hf_transfer]>=0.26.0" comfy-cli -q
+
+log_success "Dependências instaladas"
+
+# 2. Instalar ComfyUI
+log_info "[2/7] Instalando ComfyUI..."
+
+if [ -f "$COMFY_DIR/main.py" ]; then
+    log_warn "ComfyUI já existe"
+else
+    comfy --skip-prompt install --fast-deps --nvidia
+fi
+
+log_success "ComfyUI instalado"
+
+# 3. Instalar PyTorch com CUDA 12.8
+log_info "[3/7] Instalando PyTorch com CUDA 12.8..."
+pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
+log_success "PyTorch com CUDA 12.8 instalado"
+
+# 4. Instalar SageAttention
+log_info "[4/7] Instalando SageAttention..."
+if python3 -c "import torch; print(torch.cuda.is_available())" 2>/dev/null | grep -q True; then
+    pip install "https://huggingface.co/adbrasi/comfywheel/resolve/main/sageattention-2.2.0-cp311-cp311-linux_x86_64.whl" -q || log_warn "Falha ao instalar SageAttention"
+    python3 -c "import sageattention" >/dev/null 2>&1 || log_warn "SageAttention não pôde ser importado"
+    log_success "SageAttention instalado"
+else
+    log_warn "CUDA não detectado, pulando SageAttention"
+fi
+
+# 5. Baixar modelos
+log_info "[5/7] Baixando modelos..."
+
+# Criar diretórios
+mkdir -p "$MODELS_DIR"/{diffusion_models,loras,vae,vae_approx,text_encoders,clip_vision,controlnet,upscale_models,checkpoints}
+
+# Processar downloads principais
+process_downloads
+
+# Baixar Mega LoRAs
+LORA_DIR="$MODELS_DIR/loras"
+for mega_url in "${MEGA_LORAS[@]}"; do
+    download_mega "$mega_url" "$LORA_DIR"
+done
+
+# Baixar Mega Upscale
+UPSCALE_DIR="$MODELS_DIR/upscale_models"
+for mega_url in "${MEGA_UPSCALE[@]}"; do
+    download_mega "$mega_url" "$UPSCALE_DIR"
+done
+
+log_success "Downloads processados"
+
+# 6. Instalar custom nodes
+log_info "[6/7] Instalando custom nodes..."
+
+CN_DIR="$COMFY_DIR/custom_nodes"
+mkdir -p "$CN_DIR"
+
+total_nodes=${#CUSTOM_NODES[@]}
+current_node=0
+
+for repo in "${CUSTOM_NODES[@]}"; do
+    ((current_node++))
+    node_name=$(basename "$repo")
+    log_info "[$current_node/$total_nodes] Instalando: $node_name"
+    clone_repo "$repo" "$CN_DIR/$node_name"
+done
+
+log_success "Custom nodes instalados"
+
+# 7. Verificação final
+log_info "[7/7] Verificando instalação..."
+
+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
+        count=$(find "$MODELS_DIR/$dir" -type f \( -name "*.safetensors" -o -name "*.pth" -o -name "*.pt" -o -name "*.bin" \) 2>/dev/null | wc -l)
+        if [ "$count" -gt 0 ]; then 
+            log_success " $dir: $count arquivo(s)"
+        fi
+    fi
+done
+
+# Verificar e iniciar com SageAttention se disponível
+log_info "========================================="
+log_success "Instalação concluída!"
+log_info "Iniciando ComfyUI..."
+log_info "URL: http://localhost:$COMFY_PORT"
+log_info "========================================="
+
+SAGE_FLAG=""
+python3 -c "import sageattention" >/dev/null 2>&1 && SAGE_FLAG="--use-sage-attention" || true
+
+cd "$COMFY_DIR"
+exec comfy launch -- ${SAGE_FLAG:+$SAGE_FLAG} --listen "$COMFY_HOST" --port "$COMFY_PORT"