requirements.txt 382 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
transformers>=4.41.2,<=4.51.3,!=4.46.*,!=4.47.*,!=4.48.0
datasets>=2.16.0,<=3.5.0
accelerate>=0.34.0,<=1.6.0
peft>=0.14.0,<=0.15.1
chenych's avatar
chenych committed
5
trl>=0.8.6,<=0.9.6
chenych's avatar
chenych committed
6
7
tokenizers>=0.19.0,<=0.21.1
gradio>=4.38.0,<=5.25.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
protobuf
uvicorn
fastapi
sse-starlette
chenych's avatar
chenych committed
16
matplotlib>=3.7.0
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
17
fire
chenych's avatar
chenych committed
18
19
20
packaging
pyyaml
numpy<2.0.0
chenych's avatar
chenych committed
21
pydantic<=2.10.6
chenych's avatar
chenych committed
22
pandas>=2.0.0
luopl's avatar
luopl committed
23
av
chenych's avatar
chenych committed
24
librosa
luopl's avatar
luopl committed
25
tyro<0.9.0