From c4edfb7d5e591f2b2c85e2f12fdba6cb5a603e63 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Thu, 16 Mar 2023 15:33:23 +0000
Subject: [PATCH] unbump rocm5.4.2 because it does not work for me desu

---
 setup-directml.bat | 2 +-
 setup-rocm.sh      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup-directml.bat b/setup-directml.bat
index 929fb5a..3762860 100755
--- a/setup-directml.bat
+++ b/setup-directml.bat
@@ -4,7 +4,7 @@ git submodule update --remote
 python -m venv venv
 call .\venv\Scripts\activate.bat
 python -m pip install --upgrade pip
-python -m pip install torch torchvision torchaudio torch-directml
+python -m pip install torch==1.13.1 torchvision torchaudio torch-directml
 python -m pip install -r .\modules\tortoise-tts\requirements.txt
 python -m pip install -e .\modules\tortoise-tts\
 python -m pip install -r .\modules\dlas\requirements.txt
diff --git a/setup-rocm.sh b/setup-rocm.sh
index 1e010e2..3ac1cd4 100755
--- a/setup-rocm.sh
+++ b/setup-rocm.sh
@@ -7,7 +7,7 @@ python3 -m venv venv
 source ./venv/bin/activate
 python3 -m pip install --upgrade pip # just to be safe
 # ROCM
-pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
+pip3 install torch==1.13.1 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2 # 5.4.2 doesn't work for me desu
 # install requirements
 python3 -m pip install -r ./modules/tortoise-tts/requirements.txt # install TorToiSe requirements
 python3 -m pip install -e ./modules/tortoise-tts/ # install TorToiSe