transformers==4.34.0
datasets
scikit-learn
evaluate
accelerate
matplotlib
tqdm
huggingface_hub
wandb
einops
deepspeed
