torch
antialiased_cnns
pytorch_msssim
numpy~=1.19.5
torchvision~=0.10.0
pytorch_lightning
einops
huggingface_hub
