youngsheen commited on
Commit
e818756
1 Parent(s): e52682b

add requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +41 -0
requirements.txt ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # basic dependencies
2
+ torch==2.2.0
3
+ torchaudio==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
+ librosa
21
+ pytorchvideo
22
+ # misc
23
+ scikit-learn==1.2.2
24
+ huggingface_hub==0.23.4
25
+ sentencepiece==0.1.99
26
+ shortuuid
27
+ einops==0.6.1
28
+ einops-exts==0.0.4
29
+ bitsandbytes==0.43.0
30
+ pydantic>=2.0
31
+ markdown2[all]
32
+ gradio==3.50.0
33
+ gradio_client==0.6.1
34
+ httpx==0.24.1
35
+ openai==1.33.0
36
+ requests
37
+ uvicorn
38
+ fastapi
39
+ tensorboard
40
+ wandb
41
+ tabulate