configparser
sentencepiece
fire>=0.2.1
pandas<=1.5.3
scikit-learn>=0.24.2
transformers>=4.29.1
datasets==2.12.0
peft==0.3.0

[all]
unitorch[chatglm,deepspeed,diffusers,docs]

[chatglm]
protobuf==3.20.0
icetk==0.0.4
cpm_kernels==1.0.11

[deepspeed]
deepspeed==0.9.0
mpi4py==3.1.4

[diffusers]
diffusers==0.16.1

[docs]
mkdocstrings[python]>=0.18
mkdocs-material
markdown-exec
mkdocs-gen-files
mkdocs-coverage
