channels:
  - conda-forge
  - pytorch
  - nvidia
dependencies:
  - python=3.10.12
  - pip=23.2.1
  - cudatoolkit=11.8.0
  - pytorch=2.0.1
  - pytorch-cuda=11.8
  - torchvision=0.15.2
  - numpy=1.23.5
  - pandas=2.0.3
  - onnxruntime=1.15.1=py310h8a9bb50_3_cuda
  - tqdm=4.66.1
  - accelerate=0.23.0
  - terminaltables=3.1.10
  - torchmetrics=1.1.2
  - scipy==1.10.1
  - pip:
    - diffusers==0.21.1
    - omegaconf==2.3.0
    - transformers==4.33.2
    - opencv-python-headless==4.8.0.76
    - git+https://github.com/Phoveran/fastargs.git@main#egg=fastargs
    - git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
    - git+https://github.com/openai/CLIP.git@main#egg=clip