jiang-cc commited on
Commit
e5676bf
·
verified ·
1 Parent(s): 28571cb

deps: add flash-attn with ninja and build deps for source compilation

Browse files
Files changed (3) hide show
  1. packages.txt +1 -0
  2. pre-requirements.txt +4 -0
  3. requirements.txt +2 -0
packages.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ ninja-build
pre-requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ setuptools
2
+ wheel
3
+ ninja
4
+ packaging
requirements.txt CHANGED
@@ -2,6 +2,8 @@ torch
2
  torchvision
3
  transformers==4.57.3
4
  accelerate
 
5
  qwen_vl_utils
6
  Pillow
7
  spaces
 
 
2
  torchvision
3
  transformers==4.57.3
4
  accelerate
5
+ flash-attn
6
  qwen_vl_utils
7
  Pillow
8
  spaces
9
+ ninja