typing
torch
torchvision
pytorch-lightning
loguru
Pillow
huggingface_hub
numpy