torch==2.0.1
torchvision==0.15.2
transformers==4.31.0
datasets===2.12.0
peft
diffusers
sentencepiece==0.1.99
accelerate==0.21.0
bitsandbytes==0.41.0
giteapy==1.0.8
evaluate==0.4.0
tensorboard==2.13.0
scikit-learn==1.2.2
cpm_kernels
mpi4py==3.1.4