environment.yaml 734 B

12345678910111213141516171819202122232425262728293031
  1. name: ldm
  2. channels:
  3. - pytorch
  4. - defaults
  5. dependencies:
  6. - python=3.8.5
  7. - pip=20.3
  8. - cudatoolkit=11.3
  9. - pytorch=1.11.0
  10. - torchvision=0.12.0
  11. - numpy=1.19.2
  12. - pip:
  13. - albumentations==0.4.3
  14. - diffusers
  15. - opencv-python==4.1.2.30
  16. - pudb==2019.2
  17. - invisible-watermark
  18. - imageio==2.9.0
  19. - imageio-ffmpeg==0.4.2
  20. - pytorch-lightning==1.4.2
  21. - omegaconf==2.1.1
  22. - test-tube>=0.7.5
  23. - streamlit>=0.73.1
  24. - einops==0.3.0
  25. - torch-fidelity==0.3.0
  26. - transformers==4.19.2
  27. - torchmetrics==0.6.0
  28. - kornia==0.6
  29. - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
  30. - -e git+https://github.com/openai/CLIP.git@main#egg=clip
  31. - -e .