mazpie commited on
Commit
a17fea5
1 Parent(s): 88d0075

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -33,4 +33,4 @@ imageio[ffmpeg]
33
  torch==2.2.0
34
  torchvision
35
  timm==1.0.3
36
- flash_attn>=2.0.8
 
33
  torch==2.2.0
34
  torchvision
35
  timm==1.0.3
36
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.9.post1/flash_attn-2.5.9.post1+cu118torch1.12cxx11abiFALSE-cp310-cp310-linux_x86_64.whl