requirements.txt 289 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
transformers>=4.41.2,<=4.43.4
datasets>=2.16.0,<=2.20.0
accelerate>=0.30.1,<=0.32.0
peft>=0.11.1,<=0.12.0
trl>=0.8.6,<=0.9.6
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
6
gradio>=4.0.0
chenych's avatar
chenych committed
7
pandas>=2.0.0
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
8
9
10
scipy
einops
sentencepiece
chenych's avatar
chenych committed
11
tiktoken
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
12
13
14
15
16
protobuf
uvicorn
pydantic
fastapi
sse-starlette
chenych's avatar
chenych committed
17
matplotlib>=3.7.0
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
18
fire
chenych's avatar
chenych committed
19
20
21
packaging
pyyaml
numpy<2.0.0