transformers
pytorch-lightning
einops
tqdm
Pillow==7.1.2
datasets
wandb
torchmetrics
urllib3==1.22
gradio
torchvision