Skip to content

Commit 9d6fdb9

Browse files
authored
Update build-wheels-rocm.yml
Trying to use latest working version of Torch
1 parent 5d163ca commit 9d6fdb9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/build-wheels-rocm.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ jobs:
4646

4747
- name: Install Dependencies
4848
run: |
49-
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"
49+
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"
5050
pip3 install build wheel safetensors sentencepiece ninja
5151
5252
- name: Build Wheel

0 commit comments

Comments
 (0)