diff --git a/optimum/utils/import_utils.py b/optimum/utils/import_utils.py index 54f4f22709..de3bc3ed2b 100644 --- a/optimum/utils/import_utils.py +++ b/optimum/utils/import_utils.py @@ -112,6 +112,7 @@ def _is_package_available( "onnxruntime-tvm", "onnxruntime-qnn", "onnxruntime-migraphx", + "onnxruntime-windowsml", "ort-migraphx-nightly", "ort-rocm-nightly", ],