transformers
pytorch-lightning
einops
tqdm
gradio
Pillow
datasets
wandb
torchmetrics
urllib3==1.22
torchvision
scikit-learn
pandas
matplotlib
seaborn
numpy
torch
gdown