sounar commited on
Commit
20d06bb
1 Parent(s): 5be90eb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -9
requirements.txt CHANGED
@@ -1,9 +1,8 @@
1
- wheel
2
- transformers
3
- torch
4
- bitsandbytes
5
- Pillow
6
- torchvision
7
- accelerate>=0.26.0
8
- flask
9
- flash-attn
 
1
+ torch==2.5.1
2
+ transformers==4.46.3
3
+ bitsandbytes==0.44.1
4
+ Pillow==11.0.0
5
+ torchvision==0.20.1
6
+ accelerate==1.1.1
7
+ flask==3.1.0
8
+ flash-attn==2.7.0.post2