From 9d6fdb952f6705f79415364e9d85989dcda01478 Mon Sep 17 00:00:00 2001
From: turboderp <11859846+turboderp@users.noreply.github.com>
Date: Thu, 5 Oct 2023 11:36:06 +0200
Subject: [PATCH] Update build-wheels-rocm.yml

Trying to use latest working version of Torch
---
 .github/workflows/build-wheels-rocm.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/build-wheels-rocm.yml b/.github/workflows/build-wheels-rocm.yml
index d419bded..7189c47f 100644
--- a/.github/workflows/build-wheels-rocm.yml
+++ b/.github/workflows/build-wheels-rocm.yml
@@ -46,7 +46,7 @@ jobs:
           
       - name: Install Dependencies
         run: |
-          pip3 install torch\<=2.1.0 --pre --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION" --extra-index-url="https://download.pytorch.org/whl/nightly/rocm$ROCM_VERSION"
+          pip3 install torch==2.1.0.dev20230903 --pre --index-url="https://download.pytorch.org/whl/rocm$ROCM_VERSION" --extra-index-url="https://download.pytorch.org/whl/nightly/rocm$ROCM_VERSION"
           pip3 install build wheel safetensors sentencepiece ninja
         
       - name: Build Wheel