Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- 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
|
|
|
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
|