| 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 | |