Commit
·
62e0e7b
1
Parent(s):
bffa16d
Fix (script): Updated dependencies.
Browse files
quant_sdxl/env/brv_sdxl_mlperf_minimal.yml
CHANGED
|
@@ -3,6 +3,7 @@ channels:
|
|
| 3 |
- conda-forge
|
| 4 |
dependencies:
|
| 5 |
- python=3.9
|
|
|
|
| 6 |
- pip:
|
| 7 |
- torch
|
| 8 |
- accelerate==0.23.0
|
|
@@ -15,4 +16,4 @@ dependencies:
|
|
| 15 |
- tqdm
|
| 16 |
- transformers==4.33.2
|
| 17 |
- pandas==2.2.2
|
| 18 |
-
- brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@
|
|
|
|
| 3 |
- conda-forge
|
| 4 |
dependencies:
|
| 5 |
- python=3.9
|
| 6 |
+
- setuptools<70
|
| 7 |
- pip:
|
| 8 |
- torch
|
| 9 |
- accelerate==0.23.0
|
|
|
|
| 16 |
- tqdm
|
| 17 |
- transformers==4.33.2
|
| 18 |
- pandas==2.2.2
|
| 19 |
+
- brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@fb2fc8703c7492566e0748138cfb046692e9b059
|
quant_sdxl/env/brv_sdxl_mlperf_rocm_minimal.yml
CHANGED
|
@@ -3,6 +3,7 @@ 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
|
|
@@ -17,4 +18,4 @@ dependencies:
|
|
| 17 |
- tqdm
|
| 18 |
- transformers==4.33.2
|
| 19 |
- pandas==2.2.2
|
| 20 |
-
- brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@
|
|
|
|
| 3 |
- conda-forge
|
| 4 |
dependencies:
|
| 5 |
- python=3.9
|
| 6 |
+
- setuptools<70
|
| 7 |
- pip:
|
| 8 |
- --index-url https://download.pytorch.org/whl/rocm6.0
|
| 9 |
- --extra-index-url https://pypi.org/simple
|
|
|
|
| 18 |
- tqdm
|
| 19 |
- transformers==4.33.2
|
| 20 |
- pandas==2.2.2
|
| 21 |
+
- brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@fb2fc8703c7492566e0748138cfb046692e9b059
|
quant_sdxl/env/requirements.txt
CHANGED
|
@@ -8,4 +8,4 @@ torchmetrics[image]==1.2.0
|
|
| 8 |
tqdm
|
| 9 |
transformers==4.33.2
|
| 10 |
pandas==2.2.2
|
| 11 |
-
brevitas @ git+https://github.com/Xilinx/brevitas@
|
|
|
|
| 8 |
tqdm
|
| 9 |
transformers==4.33.2
|
| 10 |
pandas==2.2.2
|
| 11 |
+
brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@fb2fc8703c7492566e0748138cfb046692e9b059
|