sdxl-quant-int8 / quant_sdxl /env /brv_sdxl_mlperf_rocm_minimal.yml
nickfraser's picture
Fix (script): Updated dependencies.
62e0e7b
raw
history blame
580 Bytes
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