chenlin commited on
Commit
d6d461b
1 Parent(s): d9dadf3

add requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +29 -0
requirements.txt ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ torch==2.1.2
2
+ torchvision==0.16.2
3
+ transformers==4.37.2
4
+ tokenizers==0.15.1
5
+ sentencepiece==0.1.99
6
+ accelerate==0.21.0
7
+ peft
8
+ bitsandbytes
9
+ pydantic
10
+ markdown2[all]
11
+ numpy
12
+ scikit-learn==1.2.2
13
+ gradio==4.16.0
14
+ gradio_client==0.8.1
15
+ openai
16
+ spaces
17
+ requests
18
+ httpx==0.24.0
19
+ uvicorn
20
+ fastapi
21
+ decord
22
+ einops==0.6.1
23
+ einops-exts==0.0.4
24
+ timm==0.6.13
25
+ deepspeed==0.12.6
26
+ ninja
27
+ wandb
28
+ build
29
+ twine