From 3cd61203c59f79c2b0237eaed0815c8ebe1f00bc Mon Sep 17 00:00:00 2001
From: turboderp <11859846+turboderp@users.noreply.github.com>
Date: Fri, 2 Feb 2024 18:43:25 +0100
Subject: [PATCH] Bump Torch to 2.2.0

---
 .github/workflows/build-wheels-release-rocm.yml | 2 +-
 .github/workflows/build-wheels-release.yml      | 4 ++--
 requirements.txt                                | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build-wheels-release-rocm.yml b/.github/workflows/build-wheels-release-rocm.yml
index d88334e..029c31a 100644
--- a/.github/workflows/build-wheels-release-rocm.yml
+++ b/.github/workflows/build-wheels-release-rocm.yml
@@ -61,7 +61,7 @@ jobs:
           
       - name: Install Dependencies
         run: |
-          pip3 install torch==2.1.0 --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION"
+          pip3 install torch==2.2.0 --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION"
           pip3 install --upgrade build wheel safetensors sentencepiece ninja
 
       - name: Build Wheel
diff --git a/.github/workflows/build-wheels-release.yml b/.github/workflows/build-wheels-release.yml
index e0ad128..434a529 100644
--- a/.github/workflows/build-wheels-release.yml
+++ b/.github/workflows/build-wheels-release.yml
@@ -71,7 +71,7 @@ jobs:
           if (!(mamba list cuda)[-1].contains('cuda')) {sleep -s 10; mamba install -y 'cuda' $cudaChannels.TrimEnd().Split()}
           if (!(mamba list cuda)[-1].contains('cuda')) {throw 'CUDA Toolkit failed to install!'}
           
-          if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.0.1"} else {$torchver = "torch==2.1.0"}
+          if ([version]$env:CUDAVER -lt [version]'11.8.0') {$torchver = "torch==2.2.0"} else {$torchver = "torch==2.2.0"}
           python -m pip install $torchver --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
           
           python -m pip install --upgrade setuptools==65.5.1
@@ -160,7 +160,7 @@ jobs:
           
       - name: Install Dependencies
         run: |
-          python -m pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
+          python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu
           python -m pip install build wheel ninja
         
       - name: Build Wheel
diff --git a/requirements.txt b/requirements.txt
index b6483d8..045f250 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 pandas
 ninja
 fastparquet
-torch>=2.1.0
+torch>=2.2.0
 safetensors>=0.3.2
 sentencepiece>=0.1.97
 pygments