torch
antialiased_cnns
pytorch_msssim
numpy
torchvision>=0.11.0
pytorch_lightning
einops
huggingface_hub>=0.23.3
safetensors
wandb
datasets
tensorflow
