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