torch==1.13.1
torchvision==0.14.1
torch-fidelity==0.3.0
torchmetrics==0.11.1
einops==0.6.1
ema-pytorch==0.2.3
hydra-core==1.3.2
numpy==1.23.4
ortools==9.8.3296
lpips==0.1.4
POT==0.9.2
lion-pytorch
huggingface_hub==0.22.2
datasets==2.18.0