huangrh9 commited on
Commit
e195e70
·
verified ·
1 Parent(s): 768d4d1

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -37,4 +37,6 @@ sacrebleu
37
  hf_transfer
38
  scikit-image
39
  torch_fidelity
40
- imagesize
 
 
 
37
  hf_transfer
38
  scikit-image
39
  torch_fidelity
40
+ imagesize
41
+ # flash attention
42
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu123torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl