ClownRat commited on
Commit
e0ebef6
1 Parent(s): ff899bc

Update requirements.

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -9
requirements.txt CHANGED
@@ -1,26 +1,28 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
- huggingface_hub==0.22.2
3
- torch>=2.0.1
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
- opencv-python
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]