environment.yaml 644 B

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