Update build-wheels-rocm.yml

Trying to use latest working version of Torch
turboderp-org · Oct 5, 2023 · 9d6fdb9 · 9d6fdb9
1 parent 5d163ca
commit 9d6fdb9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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