accelerate==0.28.0
bitsandbytes==0.41.0
datasets==2.18.0
lm_eval==0.4.2
numpy==1.23.4
torch==2.2.2
tqdm==4.64.1
transformers==4.40.1
triton==2.2.0
termcolor
sentencepiece
protobuf
