requirements.txt 349 B

12345678910111213141516171819
  1. albumentations==0.4.3
  2. opencv-python
  3. pudb==2019.2
  4. imageio==2.9.0
  5. imageio-ffmpeg==0.4.2
  6. pytorch-lightning==1.4.2
  7. torchmetrics==0.6
  8. omegaconf==2.1.1
  9. test-tube>=0.7.5
  10. streamlit>=0.73.1
  11. einops==0.3.0
  12. transformers==4.19.2
  13. webdataset==0.2.5
  14. open-clip-torch==2.7.0
  15. gradio==3.13.2
  16. kornia==0.6
  17. invisible-watermark>=0.1.5
  18. streamlit-drawable-canvas==0.8.0
  19. -e .