PyYAML
numpy
pandas
torch
tensorboard
packaging
datasets
transformers
peft
bitsandbytes