torch==2.1.0
torchvision==0.16.0
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