prithivMLmods commited on
Commit
da84e63
·
verified ·
1 Parent(s): eba6924

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -6,6 +6,7 @@ timm
6
  numpy
7
  transformers
8
  transformers-stream-generator
 
9
  huggingface_hub
10
  sentencepiece
11
  peft
 
6
  numpy
7
  transformers
8
  transformers-stream-generator
9
+ git+https://github.com/Dao-AILab/flash-attention.git
10
  huggingface_hub
11
  sentencepiece
12
  peft