torch
transformers
datasets
accelerate
bitsandbytes
sentencepiece
peft
trl
tensorboardX
