# Isolated env for ControlNet (pytorch-lightning 1.5 + old torch). Needs SD v1.5 ckpt.
# Generative-augmentation baseline only; never mix with the main env.
name: seggen-controlnet
channels: [conda-forge]
dependencies:
  - python=3.10
  - pip
  - pip:
      # torch 1.12.1 (cu113 wheels bundle their own runtime):
      #   pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
      - pytorch-lightning==1.5.0
      - omegaconf==2.1.1
      - einops==0.3.0
      - transformers==4.19.2
      - open-clip-torch==2.0.2
      - gradio==3.16.2
      - opencv-python-headless
      - basicsr==1.4.2