adbrasi
/

importantFiles

Model card Files Files and versions

adbrasi commited on Nov 9, 2025

Commit

3283bab

·

verified ·

1 Parent(s): da5ccd7

Update ImageStudioTESTER.sh

Files changed (1) hide show

ImageStudioTESTER.sh +1 -1

ImageStudioTESTER.sh CHANGED Viewed

@@ -353,7 +353,7 @@ 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 nightly"
-    python -m pip install ${PIP_FLAGS} --force-reinstall --pre torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu129
 else
     log_info "GPU: ${GPU_INFO:-Não detectada}"
 fi

 if [[ "$GPU_INFO" == *"5090"* ]] || [[ "$GPU_INFO" == *"5080"* ]]; then
     log_warn "RTX 5090/5080 detectada - instalando PyTorch nightly"
+    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