requirements.txt 246 Bytes
Newer Older
luopl's avatar
luopl committed
1
2
3
4
transformers==4.41.2
datasets==2.18.0
accelerate==0.30.1
peft==0.12.0
luopl's avatar
luopl committed
5
trl>=0.8.6
luopl's avatar
luopl committed
6
7
gradio==4.14.0
pandas==2.2.2
luopl's avatar
luopl committed
8
9
10
11
12
13
14
15
16
17
18
19
20
21
scipy
einops
sentencepiece
tiktoken
protobuf
uvicorn
pydantic
fastapi
sse-starlette
matplotlib>=3.7.0
fire
packaging
pyyaml
numpy<2.0.0