einops
pytorch_lightning
streamlit
tokenizers
torch
torchvision
transformers
tqdm