Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
|
@@ -8,4 +8,6 @@ transformers==4.40.2
|
|
| 8 |
timm
|
| 9 |
|
| 10 |
torch==2.1.2
|
| 11 |
-
torchvision==0.16.2
|
|
|
|
|
|
|
|
|
| 8 |
timm
|
| 9 |
|
| 10 |
torch==2.1.2
|
| 11 |
+
torchvision==0.16.2
|
| 12 |
+
|
| 13 |
+
https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.2/flash_attn-2.6.2+cu123torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|