Skip to content

Commit 2822350

Browse files
authored
explicitly set ORT providers on all InferenceSession creations (#1262)
1 parent cd74aa2 commit 2822350

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Diff for: tests/utils/engine_mocking.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,9 @@ def __init__(
103103
with override_onnx_batch_size(
104104
model_path, batch_size, inplace=True
105105
) as batched_model_path:
106-
session = ort.InferenceSession(batched_model_path)
106+
session = ort.InferenceSession(
107+
batched_model_path, providers=["CPUExecutionProvider"]
108+
)
107109
self.input_descriptors = list(map(_to_descriptor, session.get_inputs()))
108110
self.output_descriptors = list(map(_to_descriptor, session.get_outputs()))
109111

0 commit comments

Comments
 (0)