requirements.txt 476 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
transformers>=4.41.2,<=4.49.0,!=4.46.*,!=4.47.*,!=4.48.*;python_version<'3.10'
transformers>=4.41.2,<=4.49.0,!=4.46.*,!=4.47.*,!=4.48.0;python_version>='3.10'
datasets>=2.16.0,<=3.2.0
accelerate>=0.34.0,<=1.2.1
chenych's avatar
chenych committed
5
6
peft>=0.11.1,<=0.12.0
trl>=0.8.6,<=0.9.6
chenych's avatar
chenych committed
7
tokenizers>=0.19.0,<=0.21.0
chenych's avatar
chenych committed
8
gradio>=4.38.0,<=5.21.0
chenych's avatar
chenych committed
9
pandas>=2.0.0
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
10
11
12
scipy
einops
sentencepiece
chenych's avatar
chenych committed
13
tiktoken
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
14
15
16
17
18
protobuf
uvicorn
pydantic
fastapi
sse-starlette
chenych's avatar
chenych committed
19
matplotlib>=3.7.0
Rayyyyy's avatar
V0.6.3  
Rayyyyy committed
20
fire
chenych's avatar
chenych committed
21
22
23
packaging
pyyaml
numpy<2.0.0
luopl's avatar
luopl committed
24
av
chenych's avatar
chenych committed
25
librosa
luopl's avatar
luopl committed
26
tyro<0.9.0