Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
-
torch
|
2 |
torchvision
|
3 |
#flash-attn
|
4 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
|
5 |
deepspeed
|
6 |
accelerate
|
|
|
7 |
moviepy
|
8 |
decord
|
9 |
pillow
|
|
|
1 |
+
torch==2.6.0
|
2 |
torchvision
|
3 |
#flash-attn
|
4 |
https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
|
5 |
deepspeed
|
6 |
accelerate
|
7 |
+
transformers
|
8 |
moviepy
|
9 |
decord
|
10 |
pillow
|