Maxi W commited on
Commit
0f71ee7
1 Parent(s): d707e0e

added requirements

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