codys12 commited on
Commit
3b35545
1 Parent(s): 60bf6d1

Add flash attention

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -8,3 +8,4 @@ sentencepiece
8
  spaces
9
  torch
10
  transformers
 
 
8
  spaces
9
  torch
10
  transformers
11
+ flash-attn