runninglsy commited on
Commit
2869489
·
1 Parent(s): 93cbd35

update requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -2,4 +2,5 @@ numpy==1.25.0
2
  torch==2.4.0
3
  transformers==4.46.2
4
  pillow==10.3.0
5
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu11torch2.4cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
 
 
2
  torch==2.4.0
3
  transformers==4.46.2
4
  pillow==10.3.0
5
+ # https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu11torch2.4cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
6
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post2/flash_attn-2.7.0.post2+cu11torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl