File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed
tutorials/notebooks/mct_features_notebooks/pytorch Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 2424 sed -i "/mct-quantizers/c\mct-quantizers-nightly" requirements.txt
2525 python -m pip install --upgrade pip
2626 pip install -r requirements.txt
27- pip install torch==${{ inputs.torch-version }} torchvision onnx onnxruntime onnxruntime-extensions
27+ pip install torch==${{ inputs.torch-version }} torchvision onnx onnxruntime " onnxruntime-extensions<0.14"
2828 pip install pytest pytest-mock
2929 pip check
3030 - name : Run unittests
Original file line number Diff line number Diff line change 6565 source torch_env/bin/activate
6666 python -m pip install --upgrade pip
6767 pip install -r requirements.txt
68- pip install torch==2.0.* torchvision onnx onnxruntime onnxruntime-extensions sony-custom-layers coverage pytest pytest-mock
68+ pip install torch==2.0.* torchvision onnx onnxruntime " onnxruntime-extensions<0.14" sony-custom-layers coverage pytest pytest-mock
6969
7070 - name : Run PyTorch tests (unittest)
7171 run : |
Original file line number Diff line number Diff line change 4343 {
4444 "cell_type" : " code" ,
4545 "source" : [
46- " ! pip install -q onnx onnxruntime onnxruntime-extensions"
46+ " ! pip install -q onnx onnxruntime \" onnxruntime-extensions<0.14 \" "
4747 ],
4848 "metadata" : {
4949 "id" : " qNddNV6TEsX0"
You can’t perform that action at this time.
0 commit comments