einops>=0.3.0
numpy>=1.19.5
torchvision>=0.10.0
antialiased_cnns
axial_attention
pytorch_msssim
huggingface_hub

[train]
ocf_datapipes
pytorch-lightning
