requirements.txt 335 Bytes
Newer Older
luopl's avatar
luopl committed
1
2
3
transformers>=4.41.2,<=4.46.1
datasets>=2.16.0,<=3.1.0
accelerate>=0.34.0,<=1.0.1
chenych's avatar
chenych committed
4
5
peft>=0.11.1,<=0.12.0
trl>=0.8.6,<=0.9.6
luopl's avatar
luopl committed
6
tokenizers>=0.19.0,<0.20.4
chenych's avatar
chenych committed
7
8
gradio>=4.0.0,<5.0.0
pandas>=2.0.0
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
9
10
11
scipy
einops
sentencepiece
chenych's avatar
chenych committed
12
tiktoken
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
13
14
15
16
17
protobuf
uvicorn
pydantic
fastapi
sse-starlette
chenych's avatar
chenych committed
18
matplotlib>=3.7.0
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
19
fire
chenych's avatar
chenych committed
20
21
22
packaging
pyyaml
numpy<2.0.0
luopl's avatar
luopl committed
23
24
av
tyro<0.9.0