requirements.txt 278 Bytes
Newer Older
wanglch's avatar
wanglch committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
torch
transformers==4.37.2
bitsandbytes==0.43.0
accelerate
colorama
cpm_kernels
sentencepiece
streamlit
transformers_stream_generator
peft==0.8.2
trl>=0.7.6
gradio>=3.38.0,<4.0.0
scipy
einops
sentencepiece
protobuf
jieba
rouge-chinese
nltk
uvicorn
pydantic
fastapi
sse-starlette