byh711 commited on
Commit
5bd35c3
1 Parent(s): 4d24a01

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch==2.4.0
 
2
  transformers
3
  peft==0.11.1
4
  timm
 
1
+ git+https://github.com/Dao-AILab/flash-attention.git
2
+ torch
3
  transformers
4
  peft==0.11.1
5
  timm