Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -11,4 +11,5 @@ diffusers
|
|
11 |
sentencepiece
|
12 |
timm
|
13 |
torch==2.2.0
|
14 |
-
torchvision==0.17.0
|
|
|
|
11 |
sentencepiece
|
12 |
timm
|
13 |
torch==2.2.0
|
14 |
+
torchvision==0.17.0
|
15 |
+
https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.2/flash_attn-2.8.2+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|