Skip to content

Commit 19a4b95

Browse files
committed
Revert back to torch with cuda add on
1 parent a33d0a7 commit 19a4b95

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

triton/torch_requirements.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@
1515
# H100
1616
#-----------------------------
1717
# Qwen 2.5 & Gemma3 model
18-
# --index-url https://download.pytorch.org/whl/cu126
19-
# torch==2.6.0+cu126
20-
torch==2.6.0
18+
--index-url https://download.pytorch.org/whl/cu126
19+
torch==2.6.0+cu126
20+
# torch==2.6.0
2121
torchvision==0.21.0
2222
numpy<2 # <--- numpy 2.0 is not compatible with transformers
2323

0 commit comments

Comments
 (0)