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