requirement.txt 531 B

123456789101112131415161718192021222324252627282930
  1. torch==1.12.1
  2. torchvision==0.13.1
  3. numpy==1.23.1
  4. transformers==4.26.1
  5. albumentations==1.3.0
  6. opencv-contrib-python==4.3.0.36
  7. imageio==2.9.0
  8. imageio-ffmpeg==0.4.2
  9. pytorch-lightning==1.5.0
  10. omegaconf==2.1.1
  11. test-tube>=0.7.5
  12. streamlit==1.12.1
  13. einops==0.3.0
  14. webdataset==0.2.5
  15. kornia==0.6
  16. open_clip_torch==2.0.2
  17. invisible-watermark>=0.1.5
  18. streamlit-drawable-canvas==0.8.0
  19. torchmetrics==0.6.0
  20. timm==0.6.12
  21. addict==2.4.0
  22. yapf==0.32.0
  23. prettytable==3.6.0
  24. safetensors==0.2.7
  25. basicsr==1.4.2
  26. langchain==0.0.101
  27. diffusers
  28. gradio
  29. openai
  30. accelerate