sdxl-quant-int8 / quant_sdxl /env /brv_sdxl_mlperf_minimal.yml
nickfraser's picture
Feat (script/env): Added conda environment YAMLs
bffa16d
raw
history blame
413 Bytes
name: brv_sdxl_mlperf
channels:
- conda-forge
dependencies:
- python=3.9
- 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@dev