torch>=2.3.0
torchvision
torchaudio
numpy
matplotlib
scikit-learn
opencv-python
einops>=0.7.0
omegaconf>=2.3.0
lightning>=2.1.3
transformers>=4.36.2
pudb
wandb
tensorboard
termcolor
lpips==0.1.4
ftfy
imageio
imageio-ffmpeg
pyiqa
clean-fid
albumentations
pytorch-fid
torchinfo
gradio