diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py index c67a19abbe89c7..e15500cd6f7bee 100644 --- a/machine-learning/app/models/base.py +++ b/machine-learning/app/models/base.py @@ -182,7 +182,7 @@ def provider_options_default(self) -> list[dict[str, Any]]: case "CPUExecutionProvider" | "CUDAExecutionProvider": option = {"arena_extend_strategy": "kSameAsRequested"} case "OpenVINOExecutionProvider": - option = {"disable_dynamic_shapes": True} + option = {"disable_dynamic_shapes": "True"} try: device_ids: list[str] = ort.capi._pybind_state.get_available_openvino_device_ids() log.debug(f"Available OpenVINO devices: {device_ids}") diff --git a/machine-learning/app/test_main.py b/machine-learning/app/test_main.py index 1e6a097b8480ce..f7a6afce9255e3 100644 --- a/machine-learning/app/test_main.py +++ b/machine-learning/app/test_main.py @@ -68,7 +68,7 @@ def test_sets_default_provider_options(self) -> None: encoder = OpenCLIPEncoder("ViT-B-32__openai", providers=["OpenVINOExecutionProvider", "CPUExecutionProvider"]) assert encoder.provider_options == [ - {"disable_dynamic_shapes": True}, + {"disable_dynamic_shapes": "True"}, {"arena_extend_strategy": "kSameAsRequested"}, ]