Update image_Studio.sh

image_Studio.sh

@@ -110,6 +110,9 @@ reset_state() {
 # Lista de downloads (ATUALIZADA - sem ControlNet)
 # -----------------------------
 readonly DOWNLOAD_FILES=(
+    # LoRA from Mega (PRIMEIRO para falhar rápido se houver problema)
+    "mega://https://mega.nz/file/gIRTFQSQ#no6Ay3JLE9LVRi7ib9O-Jc0CW7XmG046kCgpCzDg1tY|loras|"
+    
     # Checkpoint mantido
     "https://civitai.com/api/download/models/2122278?type=Model&format=SafeTensor&size=pruned&fp=fp16|checkpoints|raehoshiIllustXL_v60.safetensors"
     
@@ -121,9 +124,6 @@ readonly DOWNLOAD_FILES=(
     "https://civitai.com/api/download/models/1114313?type=Model&format=SafeTensor|loras|"
     "https://civitai.com/api/download/models/1780244?type=Model&format=SafeTensor|loras|"
     
-    # LoRA from Mega (nome automático pelo megadl)
-    "mega://https://mega.nz/file/gIRTFQSQ#no6Ay3JLE9LVRi7ib9O-Jc0CW7XmG046kCgpCzDg1tY|loras|"
-    
     # Upscale models - Coleção completa
     "https://huggingface.co/Kim2091/AnimeSharp/resolve/main/4x-AnimeSharp.pth|upscale_models|4x-AnimeSharp.pth"
     "https://huggingface.co/Kim2091/AnimeSharpV3/resolve/main/2x-AnimeSharpV3.pth|upscale_models|2x-AnimeSharpV3.pth"
@@ -219,24 +219,16 @@ download_mega() {
     # Remover prefixo mega:// se houver
     url="${url#mega://}"
     
-    log_info "Baixando do Mega para $target_dir"
+    log_info "Executando: megadl '$url'"
     
-    # Usar megadl simples como você sempre usa
+    # Usar megadl com timeout
     cd "$target_dir"
-    if megadl "$url"; then
-        # Se especificou nome, renomear
-        if [ -n "$filename" ]; then
-            # Encontrar o arquivo baixado (o mais recente)
-            local downloaded=$(ls -t | head -1)
-            if [ -n "$downloaded" ] && [ "$downloaded" != "$filename" ]; then
-                mv "$downloaded" "$filename"
-            fi
-        fi
+    if timeout 300 megadl "$url"; then
         log_success "Download do Mega concluído"
         cd - >/dev/null
         return 0
     else
-        log_error "Falha no download do Mega"
+        log_error "Mega falhou ou timeout (5 min)"
         cd - >/dev/null
         return 1
     fi
@@ -364,16 +356,21 @@ process_downloads() {
     local failed_count=0
     
     log_info "Total de downloads: $total"
+    log_info "Iniciando loop de downloads..."
     
     for entry in "${DOWNLOAD_FILES[@]}"; do
         ((current++))
         
+        log_info "[$current/$total] Processando entrada..."
+        
         # Verificar se já foi baixado
         if is_download_complete "$entry"; then
             log_success "[$current/$total] Já baixado, pulando..."
             continue
         fi
         
+        log_info "[$current/$total] Preparando download..."
+        
         IFS='|' read -r url type filename <<< "$entry"
         
         # Limpar espaços
@@ -381,6 +378,9 @@ process_downloads() {
         type=$(echo "$type" | xargs)
         filename=$(echo "$filename" | xargs)
         
+        log_info "[$current/$total] URL: ${url:0:60}..."
+        log_info "[$current/$total] Tipo: $type"
+        
         # Determinar diretório de destino
         local target_dir="$MODELS_DIR"
         case "$type" in
@@ -594,8 +594,9 @@ main() {
         local gpu_info=$(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null || echo "")
         
         if [[ "$gpu_info" == *"5090"* ]] || [[ "$gpu_info" == *"5080"* ]]; then
-            log_warn "RTX ${gpu_info} detectada - corrigindo PyTorch"
-            # Instalar torch, torchvision e torchaudio JUNTOS em um comando só
+            log_warn "RTX ${gpu_info} detectada - instalando PyTorch nightly"
+            # FORÇAR reinstalação do PyTorch nightly
+            pip uninstall -y torch torchvision torchaudio 2>/dev/null || true
             pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
             log_success "PyTorch nightly instalado"
         else
@@ -610,7 +611,10 @@ main() {
         log_info "[5/6] Baixando modelos..."
         mkdir -p "$MODELS_DIR"/{diffusion_models,loras,vae,vae_approx,text_encoders,clip_vision,upscale_models,checkpoints,ultralytics/bbox}
         
+        log_info "Chamando process_downloads..."
         process_downloads
+        log_info "process_downloads terminou"
+        
         log_success "Downloads processados"
         save_state "5"
     fi