QinOwen commited on
Commit
0e20ffa
1 Parent(s): 9e782de

add-requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +30 -0
requirements.txt ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ decord==0.6.0
2
+ einops==0.3.0
3
+ imageio==2.9.0
4
+ numpy==1.24.2
5
+ omegaconf==2.1.1
6
+ opencv_python
7
+ pandas==2.0.0
8
+ Pillow==9.5.0
9
+ pytorch_lightning==2.3.1
10
+ PyYAML==6.0
11
+ setuptools==65.6.3
12
+ tqdm==4.65.0
13
+ transformers==4.25.1
14
+ moviepy==1.0.3
15
+ av==12.2.0
16
+ gradio
17
+ timm==1.0.7
18
+ scikit-learn==1.5.0
19
+ open_clip_torch==2.22.0
20
+ kornia==0.7.3
21
+ albumentations==1.3.1
22
+ peft==0.11.1
23
+ bitsandbytes==0.42.0
24
+ accelerate==0.31.0
25
+ inflect==7.3.0
26
+ wandb==0.17.3
27
+ ipdb==0.13.13
28
+ huggingface-hub==0.23.4
29
+ gradio
30
+ -e git+https://github.com/tgxs002/HPSv2.git#egg=hpsv2