Update wanFullForTest.sh

wanFullForTest.sh

@@ -85,6 +85,12 @@ readonly MEGA_LORAS=(
     "https://mega.nz/file/wcAFhRJZ#ifZcckts0wPq1FG693tYLkpFrvjXvab1ZQeL6DuL1T8"
 )
 
+# Mega.nz Upscale models
+readonly MEGA_UPSCALE=(
+    "https://mega.nz/file/qdQwAJSC#rPY2Hg4x-_RbHiHUBFW-YU8xZUqPpq5Gpl1uPFiJvW4"
+    "https://mega.nz/file/KIh2marA#CehVNXxQhGfp1bqUL0K1YPWNqrCUoLPCZGq2W6XqmWo"
+)
+
 # Custom nodes para instalar
 readonly CUSTOM_NODES=(
     "https://github.com/kijai/ComfyUI-Florence2"
@@ -306,6 +312,8 @@ process_downloads() {
             clip_vision) target_dir="$MODELS_DIR/clip_vision" ;;
             controlnet) target_dir="$MODELS_DIR/controlnet" ;;
             upscale_models) target_dir="$MODELS_DIR/upscale_models" ;;
+            tensorrt) target_dir="$MODELS_DIR/tensorrt" ;;
+            tensorrt_rife) target_dir="$MODELS_DIR/tensorrt/rife" ;;
             *) target_dir="$MODELS_DIR/$type" ;;
         esac
         
@@ -503,18 +511,25 @@ mkdir -p "$LORA_DIR"
 
 if command_exists megadl; then
     for mega_url in "${MEGA_LORAS[@]}"; do
-        log_info "Processando: $mega_url"
+        log_info "Processando LoRA: $mega_url"
         download_mega "$mega_url" "$LORA_DIR" || log_warn "Falha ao baixar: $mega_url"
     done
-    log_success "Downloads Mega.nz concluídos"
+    log_success "LoRAs do Mega.nz concluídos"
+
+    # Baixar Upscale models do Mega.nz
+    log_info "Baixando Upscale models do Mega.nz..."
+    UPSCALE_DIR="$MODELS_DIR/upscale_models"
+    mkdir -p "$UPSCALE_DIR"
+
+    for mega_url in "${MEGA_UPSCALE[@]}"; do
+        log_info "Processando Upscale: $mega_url"
+        download_mega "$mega_url" "$UPSCALE_DIR" || log_warn "Falha ao baixar: $mega_url"
+    done
+    log_success "Upscale models do Mega.nz concluídos"
 else
     log_warn "megatools não disponível, pulando downloads do Mega.nz"
 fi
 
-# TODO: Adicionar download da pasta Mega para upscale_models
-# https://mega.nz/folder/rdpkjZzC#eUXPed_vntJKLrB0wpeJ-w
-# Nota: megadl não suporta pastas diretamente, requer links individuais
-
 # [9] Instalar custom nodes
 log_info "[9/10] Instalando custom nodes..."
 CN_DIR="$COMFY_DIR/custom_nodes"