transformers
numpy
scipy
datasets
pyarrow
tqdm
torch
bitsandbytes