Skip to content

Commit d7ad4a9

Browse files
committed
Fix PyTorch requirement for Linux
1 parent 3f548eb commit d7ad4a9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,5 @@ bitsandbytes==0.42.0; platform_system != "Windows"
1111
https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.2.post2-py3-none-win_amd64.whl; platform_system == "Windows"
1212

1313
# PyTorch
14-
--find-links https://download.pytorch.org/whl/torch_stable.html
1514
torch==2.1.2; platform_system != "Windows"
16-
torch==2.1.2+cu121; platform_system == "Windows"
15+
https://download.pytorch.org/whl/cu121/torch-2.1.2%2Bcu121-cp311-cp311-win_amd64.whl; platform_system == "Windows"

0 commit comments

Comments
 (0)