Spaces:
Running
on
A100
Running
on
A100
Commit
·
f0a6126
1
Parent(s):
e933e47
use torch 2.8.0
Browse files- requirements.txt +6 -4
requirements.txt
CHANGED
|
@@ -1,7 +1,9 @@
|
|
| 1 |
-
torch
|
| 2 |
-
torchvision
|
| 3 |
-
torchdata
|
| 4 |
-
torchao
|
|
|
|
|
|
|
| 5 |
|
| 6 |
accelerate==1.9.0
|
| 7 |
av==15.0.0
|
|
|
|
| 1 |
+
torch==2.8.0
|
| 2 |
+
torchvision==0.23.0
|
| 3 |
+
torchdata==0.11.0
|
| 4 |
+
torchao==0.12.0
|
| 5 |
+
|
| 6 |
+
flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
| 7 |
|
| 8 |
accelerate==1.9.0
|
| 9 |
av==15.0.0
|