torch
datasets
accelerate
bitsandbytes
transformers
sentencepiece
numpy