torch==2.0.1
tokenizers==0.14.0
transformers==4.35.0
accelerate==0.24.1
numpy==1.26.4
attrdict
tqdm

datasets
tensorboardX