configparser
pillow
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[deepspeed,diffusers,quantization]

[deepspeed]
deepspeed==0.9.0
mpi4py==3.1.4

[diffusers]
diffusers>=0.17.1

[quantization]
bitsandbytes>=0.41.1
