File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -135,7 +135,7 @@ jobs:
135135
136136 - name : Install python test dependencies
137137 run : |
138- pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
138+ pip install pyyaml pandas scipy ' numpy==1.26.4' psutil pyre_extensions torchrec
139139
140140 - name : Install transformers package
141141 if : ${{ inputs.suite == 'huggingface' }}
Original file line number Diff line number Diff line change @@ -165,7 +165,7 @@ jobs:
165165 - name : Install python test dependencies
166166 run : |
167167 .venv\Scripts\activate.ps1
168- pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
168+ pip install pyyaml pandas scipy ' numpy==1.26.4' psutil pyre_extensions torchrec
169169
170170 - name : Install transformers package
171171 if : inputs.suite == 'all' || inputs.suite == 'huggingface'
Original file line number Diff line number Diff line change @@ -468,7 +468,7 @@ run_inductor_tests() {
468468 git checkout $rev
469469 )
470470
471- pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
471+ pip install pyyaml pandas scipy ' numpy==1.26.4 ' psutil pyre_extensions torchrec
472472
473473 # TODO: Find the fastest Hugging Face model
474474 ZE_AFFINITY_MASK=0 python pytorch/benchmarks/dynamo/huggingface.py --accuracy --float32 -dxpu -n10 --no-skip --dashboard --inference --freezing --total-partitions 1 --partition-id 0 --only AlbertForMaskedLM --backend=inductor --timeout=4800 --output=$( pwd -P) /inductor_log.csv
You can’t perform that action at this time.
0 commit comments