requirements.txt

torch>=2.0.0
torchvision>=0.15.0
Pillow>=9.0.0
opencv-python>=4.7.0
imageio>=2.31.0
matplotlib>=3.7.0
omegaconf>=2.3.0
ftfy>=6.1.1
regex>=2023.0.0
tqdm>=4.65.0
PyYAML>=6.0
einops>=0.6.0
pytorch-lightning