File tree 2 files changed +2
-2
lines changed
2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -124,7 +124,7 @@ jobs:
124
124
conda activate ci
125
125
126
126
echo '::group::Install ONNX'
127
- pip install --progress-bar=off onnx onnxruntime!=1.16.0
127
+ pip install --progress-bar=off onnx onnxruntime
128
128
echo '::endgroup::'
129
129
130
130
echo '::group::Install testing utilities'
Original file line number Diff line number Diff line change @@ -79,7 +79,7 @@ def to_numpy(tensor):
79
79
inputs = list (map (to_numpy , inputs ))
80
80
outputs = list (map (to_numpy , outputs ))
81
81
82
- ort_session = onnxruntime .InferenceSession (onnx_io .getvalue ())
82
+ ort_session = onnxruntime .InferenceSession (onnx_io .getvalue (), providers = onnxruntime . get_available_providers () )
83
83
# compute onnxruntime output prediction
84
84
ort_inputs = {ort_session .get_inputs ()[i ].name : inpt for i , inpt in enumerate (inputs )}
85
85
ort_outs = ort_session .run (None , ort_inputs )
You can’t perform that action at this time.
0 commit comments