Skip to content

Commit 85f3022

Browse files
committed
fix: change unsloth torch version from torch260 to torch251 for cublas12 build
The torch260 package is not available on PyPI for cu121-ampere. Changed to torch251 which is the latest available version that is compatible.
1 parent ae5ed50 commit 85f3022

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

backend/python/unsloth/requirements-cublas12.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
--extra-index-url https://download.pytorch.org/whl/cu121
2-
unsloth[cu121-ampere-torch260]
2+
unsloth[cu121-ampere-torch251]
33
torch
44
transformers
55
datasets

0 commit comments

Comments
 (0)