Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.
Browse files- requirements.txt +11 -9
requirements.txt
CHANGED
@@ -1,26 +1,28 @@
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
2 |
-
|
3 |
-
torch
|
4 |
-
torchvision>=0.15.2
|
5 |
transformers==4.37.2
|
6 |
tokenizers==0.15.1
|
7 |
-
sentencepiece==0.1.99
|
8 |
-
shortuuid
|
9 |
deepspeed==0.13.1
|
10 |
accelerate==0.21.0
|
11 |
peft==0.4.0
|
|
|
|
|
|
|
12 |
decord==0.6.0
|
13 |
-
pytorchvideo==0.1.5
|
14 |
imageio==2.34.0
|
15 |
imageio-ffmpeg==0.4.9
|
|
|
|
|
16 |
moviepy==1.0.3
|
17 |
scenedetect==0.6.3
|
18 |
-
|
19 |
-
numpy
|
20 |
scikit-learn==1.2.2
|
|
|
|
|
|
|
21 |
einops==0.6.1
|
22 |
einops-exts==0.0.4
|
23 |
-
timm==0.6.13
|
24 |
bitsandbytes==0.41.0
|
25 |
pydantic<2,>=1
|
26 |
markdown2[all]
|
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
2 |
+
# basic dependencies
|
3 |
+
torch==2.0.1
|
|
|
4 |
transformers==4.37.2
|
5 |
tokenizers==0.15.1
|
|
|
|
|
6 |
deepspeed==0.13.1
|
7 |
accelerate==0.21.0
|
8 |
peft==0.4.0
|
9 |
+
timm==0.6.13
|
10 |
+
numpy
|
11 |
+
# data processing
|
12 |
decord==0.6.0
|
|
|
13 |
imageio==2.34.0
|
14 |
imageio-ffmpeg==0.4.9
|
15 |
+
torchvision>=0.15.2
|
16 |
+
pytorchvideo==0.1.5
|
17 |
moviepy==1.0.3
|
18 |
scenedetect==0.6.3
|
19 |
+
# misc
|
|
|
20 |
scikit-learn==1.2.2
|
21 |
+
huggingface_hub==0.22.2
|
22 |
+
sentencepiece==0.1.99
|
23 |
+
shortuuid
|
24 |
einops==0.6.1
|
25 |
einops-exts==0.0.4
|
|
|
26 |
bitsandbytes==0.41.0
|
27 |
pydantic<2,>=1
|
28 |
markdown2[all]
|