Update ImageStudioArrakis2.sh

ImageStudioArrakis2.sh

@@ -104,7 +104,6 @@ CUSTOM_NODES=(
     "https://github.com/KoreTeknology/ComfyUI-Universal-Styler"
     "https://github.com/WASasquatch/was-node-suite-comfyui"
     "https://github.com/chflame163/ComfyUI_LayerStyle"
-    "https://github.com/newtextdoc1111/ComfyUI-Autocomplete-Plus"
     "https://github.com/pythongosssss/ComfyUI-WD14-Tagger"
     "https://github.com/ltdrdata/ComfyUI-Impact-Pack"
     "https://github.com/pythongosssss/ComfyUI-Custom-Scripts"
@@ -119,7 +118,6 @@ CUSTOM_NODES=(
     "https://github.com/Cezarsaint/rand0micoUploaderLoven"
     "https://github.com/adbrasi/pixivmosaic"
     "https://github.com/adbrasi/futfilter"
-    "https://github.com/Artificial-Sweetener/comfyui-WhiteRabbit"
     "https://github.com/shiimizu/ComfyUI_smZNodes"
     "https://github.com/CoreyCorza/ComfyUI-CRZnodes"
     "https://github.com/MoonGoblinDev/Civicomfy"
@@ -293,6 +291,20 @@ fi
 
 log_success "ComfyUI instalado"
 
+# 4. Corrigir PyTorch para RTX 5090 se detectado
+log_info "[4/6] Verificando GPU..."
+
+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 5090/5080 detectada - instalando PyTorch novo"
+    python -m pip install ${PIP_FLAGS} --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
+else
+    log_info "GPU: ${GPU_INFO:-Não detectada}"
+fi
+
+log_success "PyTorch configurado"
+
 # 5. Baixar modelos
 log_info "[5/6] Baixando modelos..."