Joe6636564 commited on
Commit
f914beb
·
verified ·
1 Parent(s): f75af51

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -5
requirements.txt CHANGED
@@ -1,6 +1,9 @@
1
  flask
2
- pillow
3
- torch
4
- transformers
5
- accelerate
6
- safetensors
 
 
 
 
1
  flask
2
+ flash_attn==2.5.8
3
+ numpy==1.24.4
4
+ Pillow==10.3.0
5
+ Requests==2.31.0
6
+ torch==2.3.0
7
+ torchvision==0.18.0
8
+ transformers==4.43.0
9
+ accelerate==0.30.0