torch==2.2.1 torchvision==0.17.1 torchmetrics==0.6.0 opencv-python==4.6.0.66 pytorch-lightning==1.4.9 omegaconf==2.3.0 einops==0.6.1 kornia==0.6.12 gradio==3.41.2 gradio_client==0.5.0