name: ldm
channels:
  - pytorch
  - conda-forge
dependencies:
  - python==3.10.5
  - pip==22.2.2

  - pytorch
  - torchvision

  - albumentations==1.2.1
  - coloredlogs==15.0.1
  - einops==0.4.1
  - grpcio==1.46.4
  - humanfriendly==10.0
  - imageio==2.21.2
  - imageio-ffmpeg==0.4.7
  - imgaug==0.4.0
  - kornia==0.6.7
  - mpmath==1.2.1
  - nomkl
  - numpy==1.23.2
  - omegaconf==2.1.1
  - onnx==1.12.0
  - onnxruntime==1.12.1
  - pudb==2022.1
  - pytorch-lightning==1.6.5
  - scipy==1.9.1
  - streamlit==1.12.2
  - sympy==1.10.1
  - tensorboard==2.9.0
  - torchmetrics==0.9.3
  - antlr4-python3-runtime=4.8
  - pip:
    - opencv-python==4.6.0.66
    - realesrgan==0.2.5.0
    - test-tube==0.7.5
    - transformers==4.21.2
    - torch-fidelity==0.3.0
    - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
    - -e git+https://github.com/openai/CLIP.git@main#egg=clip
    - -e git+https://github.com/TencentARC/GFPGAN#egg=GFPGAN
    - -e git+https://github.com/xinntao/Real-ESRGAN#egg=realesrgan
    - -e .
variables:
  PYTORCH_ENABLE_MPS_FALLBACK: 1