KaiChen1998 commited on
Commit
6f248e7
·
1 Parent(s): 997d449

add flash attention dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -7,6 +7,7 @@ accelerate==0.33.0
7
  einops==0.6.1
8
  einops-exts==0.0.4
9
  timm==0.6.13
 
10
  scipy
11
  gradio
12
 
 
7
  einops==0.6.1
8
  einops-exts==0.0.4
9
  timm==0.6.13
10
+ flash_attn
11
  scipy
12
  gradio
13