Skip to content

Commit e07936e

Browse files
authored
Merge pull request #30 from jhc13/bitsandbytes-windows
Support loading in 4-bit on Windows
2 parents 7b4cd34 + d7ad4a9 commit e07936e

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

requirements.txt

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,15 @@
11
accelerate==0.25.0
2-
bitsandbytes==0.41.3.post2
32
imagesize==1.4.1
43
Pillow==10.2.0
54
pyparsing==3.1.1
65
PySide6==6.6.1
7-
scipy==1.11.4
8-
torch==2.1.2
96
transformers==4.36.2
7+
8+
# bitsandbytes
9+
scipy==1.11.4
10+
bitsandbytes==0.42.0; platform_system != "Windows"
11+
https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.2.post2-py3-none-win_amd64.whl; platform_system == "Windows"
12+
13+
# PyTorch
14+
torch==2.1.2; 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)