sdxl-quant-int8 / quant_sdxl /env /brv_sdxl_mlperf_minimal.yml
GiusFra's picture
Updates to minimal quantization script. (#1)
72eb84b verified
raw
history blame
468 Bytes
name: brv_sdxl_mlperf
channels:
- conda-forge
dependencies:
- python=3.9
- setuptools<70
- pip:
- 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