name: brv_sdxl_mlperf_rocm
channels:
  - conda-forge
dependencies:
  - python=3.9
  - setuptools<70
  - pip:
    - --index-url https://download.pytorch.org/whl/rocm6.0
    - --extra-index-url https://pypi.org/simple
    - torch
    - accelerate==0.23.0
    - diffusers==0.21.2
    - open-clip-torch==2.7.0
    - opencv-python==4.8.1.78
    - pycocotools==2.0.7
    - scipy==1.9.1
    - torchmetrics[image]==1.2.0
    - tqdm
    - transformers==4.33.2
    - pandas==2.2.2
    - brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@fb2fc8703c7492566e0748138cfb046692e9b059