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 gradio_client==0.2.7