From 012b987ab8d9274ba988fe6c3939696c373f5839 Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Fri, 14 Jun 2024 13:10:17 +0100
Subject: [PATCH] Force PyTorch 2.2.1

---
 atsetup.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/atsetup.sh b/atsetup.sh
index 070ef83..0b458da 100644
--- a/atsetup.sh
+++ b/atsetup.sh
@@ -325,7 +325,7 @@ install_custom_standalone() {
     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."
-    pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinstall --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
     echo "    Installing additional requirements."
     echo