torch==2.0.1
torchvision==0.15.2
transformers==4.29.1
datasets==2.8.0
peft
diffusers
sentencepiece==0.1.99
accelerate>=0.20.3
bitsandbytes>=0.39.0
giteapy==1.0.8
evaluate==0.4.0
tensorboard==2.13.0
scikit-learn==1.2.2
cpm_kernels
mpi4py==3.1.4