nickfraser commited on
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@dev
 
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@dev
 
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@dev
 
8
  tqdm
9
  transformers==4.33.2
10
  pandas==2.2.2
11
+ brevitas[export] @ git+https://github.com/Xilinx/brevitas.git@fb2fc8703c7492566e0748138cfb046692e9b059