teowu commited on
Commit
56c0224
1 Parent(s): a553a03

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  icecream
2
  markdown2
 
3
  pydantic
4
  accelerate==0.21.0
5
  transformers==4.36.2
 
1
  icecream
2
  markdown2
3
+ flash_attn
4
  pydantic
5
  accelerate==0.21.0
6
  transformers==4.36.2