From 645c0d3a6796ea855e8d56a1184c30ccf56f18a9 Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Fri, 29 Mar 2024 00:42:54 +0000
Subject: [PATCH] Update atsetup.bat

force pytorch upgrade
---
 atsetup.bat | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/atsetup.bat b/atsetup.bat
index b4cfbfc..42510e9 100644
--- a/atsetup.bat
+++ b/atsetup.bat
@@ -383,7 +383,7 @@ echo     Downloading and installing PyTorch. This step can take a long time
 echo     depending on your internet connection and hard drive speed. Please
 echo     be patient.
 echo.
-pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
+pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu121
 echo Installing other requirements.
 echo.
 pip install -r system\requirements\requirements_standalone.txt
@@ -524,7 +524,7 @@ echo     Downloading and installing PyTorch. This step can take a long time
 echo     depending on your internet connection and hard drive speed. Please
 echo     be patient.
 echo.
-pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
+pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu1
 echo Installing other requirements.
 echo.
 pip install -r system\requirements\requirements_standalone.txt