Skip to content

Commit 0f820f3

Browse files
committed
Installing torch and triton first so that gradlib is built compatible
1 parent 3103abb commit 0f820f3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

.github/workflows/publish.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,11 @@ jobs:
5252
fail-fast: false
5353

5454
steps:
55+
- name: Prepare
56+
run: |
57+
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2
58+
pip3 install -U triton
59+
5560
- name: Checkout
5661
uses: actions/checkout@eef61447b9ff4aafe5dcd4e0bbf5d482be7e7871 # v4.2.1
5762

0 commit comments

Comments
 (0)