diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index adf8ed8b0694..01747f32685d 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -140,6 +140,7 @@ def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> Tuple[b "onnxruntime-rocm", "onnxruntime-training", "onnxruntime-vitisai", + "onnxruntime-winml", ) _onnxruntime_version = None # For the metadata, we have to look for both onnxruntime and onnxruntime-x