torch>=1.12.1
torchvision>=0.13.1
torchmetrics>=0.6.0
opencv-python>=4.5.5.64
pytorch-lightning>=1.4.2
omegaconf>=2.1.1
einops>=0.3.0
kornia>=0.6
gradio==3.41.2
gradio_client==0.5.0