albumentations==1.3.0 opencv-python pudb==2019.2 prefetch_generator imageio==2.9.0 imageio-ffmpeg==0.4.2 torchmetrics==0.6 omegaconf==2.1.1 test-tube>=0.7.5 streamlit>=0.73.1 einops==0.3.0 transformers==4.19.2 webdataset==0.2.5 open-clip-torch==2.7.0 gradio==3.11 datasets -e .