2023-05-29 21:53:02 +08:00
|
|
|
torch>=1.13.1
|
2023-11-06 19:01:21 +08:00
|
|
|
transformers>=4.31.0,<4.35.0
|
2023-06-12 15:48:48 +08:00
|
|
|
datasets>=2.12.0
|
2023-07-21 16:48:34 +08:00
|
|
|
accelerate>=0.21.0
|
2023-11-07 16:13:36 +08:00
|
|
|
peft>=0.6.0
|
2023-11-07 16:17:22 +08:00
|
|
|
trl==0.7.2
|
2023-11-06 19:01:21 +08:00
|
|
|
gradio>=3.38.0,<4.0.0
|
2023-08-09 00:26:11 +08:00
|
|
|
scipy
|
2023-07-05 22:42:31 +08:00
|
|
|
sentencepiece
|
2023-09-11 17:31:34 +08:00
|
|
|
protobuf
|
2023-08-03 15:53:32 +08:00
|
|
|
tiktoken
|
2023-10-09 20:02:50 +08:00
|
|
|
fire
|
2023-05-29 21:53:02 +08:00
|
|
|
jieba
|
2023-07-05 23:03:58 +08:00
|
|
|
rouge-chinese
|
2023-05-29 21:53:02 +08:00
|
|
|
nltk
|
2023-06-23 00:17:05 +08:00
|
|
|
uvicorn
|
2023-11-06 19:01:21 +08:00
|
|
|
pydantic
|
|
|
|
fastapi
|
2023-07-05 23:03:58 +08:00
|
|
|
sse-starlette
|
2023-07-15 16:54:28 +08:00
|
|
|
matplotlib
|