torch
Pillow
torchvision
numpy
pytorch-lightning==2.2.5