Aliayub1995 commited on
Commit
4c4dee6
1 Parent(s): 0e622c7

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +39 -4
requirements.txt CHANGED
@@ -1,4 +1,39 @@
1
- # requirements.txt
2
-
3
- # This file triggers the setup.sh script
4
- -r install.sh
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ # basic dependencies
3
+ torch==2.2.0
4
+ torchvision==0.17.0
5
+ transformers==4.42.3
6
+ tokenizers==0.19.1
7
+ deepspeed==0.13.1
8
+ accelerate==0.26.1
9
+ peft==0.4.0
10
+ timm==1.0.3
11
+ numpy==1.24.4
12
+ # data processing
13
+ decord==0.6.0
14
+ imageio==2.34.0
15
+ imageio-ffmpeg==0.4.9
16
+ moviepy==1.0.3
17
+ scenedetect==0.6.3
18
+ opencv-python==4.6.0.66
19
+ pysubs2
20
+ # misc
21
+ scikit-learn==1.2.2
22
+ huggingface_hub==0.23.4
23
+ sentencepiece==0.1.99
24
+ shortuuid
25
+ einops==0.6.1
26
+ einops-exts==0.0.4
27
+ bitsandbytes==0.43.0
28
+ pydantic>=2.0
29
+ markdown2[all]
30
+ gradio==3.50.0
31
+ gradio_client==0.6.1
32
+ httpx==0.24.1
33
+ requests
34
+ openai
35
+ uvicorn
36
+ fastapi
37
+ tensorboard
38
+ wandb
39
+ tabulate