LinB203 commited on
Commit
bb48a96
β€’
1 Parent(s): 61f3f56
Files changed (1) hide show
  1. requirements.txt +30 -0
requirements.txt ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ pip
3
+ einops
4
+ fastapi
5
+ gradio==3.37.0
6
+ markdown2[all]
7
+ numpy
8
+ requests
9
+ sentencepiece
10
+ tokenizers>=0.12.1
11
+ torch==2.0.1
12
+ torchvision==0.15.2
13
+ uvicorn
14
+ wandb
15
+ shortuuid
16
+ httpx==0.24.0
17
+ deepspeed==0.9.5
18
+ peft==0.4.0
19
+ transformers==4.31.0
20
+ accelerate==0.21.0
21
+ bitsandbytes==0.41.0
22
+ scikit-learn==1.2.2
23
+ sentencepiece==0.1.99
24
+ einops==0.6.1
25
+ einops-exts==0.0.4
26
+ timm==0.6.13
27
+ gradio_client==0.7.0
28
+ decord==0.6.0
29
+ opencv-python==4.7.0.72
30
+ pytorchvideo @ git+https://github.com/facebookresearch/pytorchvideo.git@28fe037d212663c6a24f373b94cc5d478c8c1a1d