channels:
  - pytorch
  - nvidia
  - defaults
dependencies:
  - python=3.9
  - pytorch=2.3
  - torchvision
  - torchaudio
  - pytorch-cuda=12.1
  - pip
  - pip:
    - diffusers==0.20.0
    - gradio==3.48.0
    - imageio==2.34.1
    - numpy==1.26.4
    - opencv-python==4.9.0.80
    - transformers==4.40.2