numpy
transformers
torch==2.0.0
sentencepiece
tokenizers
accelerate
deepspeed
