torch
transformers
accelerate
datasets
scikit-learn
tqdm
numpy
bitsandbytes
