torch
torchvision
tqdm
wandb
transformers==4.30.2