Commit
•
bffa16d
1
Parent(s):
e01fcf2
Feat (script/env): Added conda environment YAMLs
Browse files
quant_sdxl/env/brv_sdxl_mlperf_minimal.yml
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: brv_sdxl_mlperf
|
| 2 |
+
channels:
|
| 3 |
+
- conda-forge
|
| 4 |
+
dependencies:
|
| 5 |
+
- python=3.9
|
| 6 |
+
- pip:
|
| 7 |
+
- torch
|
| 8 |
+
- accelerate==0.23.0
|
| 9 |
+
- diffusers==0.21.2
|
| 10 |
+
- open-clip-torch==2.7.0
|
| 11 |
+
- opencv-python==4.8.1.78
|
| 12 |
+
- pycocotools==2.0.7
|
| 13 |
+
- scipy==1.9.1
|
| 14 |
+
- torchmetrics[image]==1.2.0
|
| 15 |
+
- tqdm
|
| 16 |
+
- transformers==4.33.2
|
| 17 |
+
- pandas==2.2.2
|
| 18 |
+
- brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@dev
|
quant_sdxl/env/brv_sdxl_mlperf_rocm_minimal.yml
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: brv_sdxl_mlperf_rocm
|
| 2 |
+
channels:
|
| 3 |
+
- conda-forge
|
| 4 |
+
dependencies:
|
| 5 |
+
- python=3.9
|
| 6 |
+
- pip:
|
| 7 |
+
- --index-url https://download.pytorch.org/whl/rocm6.0
|
| 8 |
+
- --extra-index-url https://pypi.org/simple
|
| 9 |
+
- torch
|
| 10 |
+
- accelerate==0.23.0
|
| 11 |
+
- diffusers==0.21.2
|
| 12 |
+
- open-clip-torch==2.7.0
|
| 13 |
+
- opencv-python==4.8.1.78
|
| 14 |
+
- pycocotools==2.0.7
|
| 15 |
+
- scipy==1.9.1
|
| 16 |
+
- torchmetrics[image]==1.2.0
|
| 17 |
+
- tqdm
|
| 18 |
+
- transformers==4.33.2
|
| 19 |
+
- pandas==2.2.2
|
| 20 |
+
- brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@dev
|