Update import_utils.py (huggingface#10329)

Lakshaysharma048 · web-flow · commit b0ff822ed385 · 2025-04-02T20:47:10.000+01:00
added onnxruntime-vitisai for custom build onnxruntime pkg
diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py
@@ -109,6 +109,7 @@ def _is_package_available(pkg_name: str):
         "onnxruntime-rocm",
         "onnxruntime-migraphx",
         "onnxruntime-training",
+        "onnxruntime-vitisai",
     )
     _onnxruntime_version = None
     # For the metadata, we have to look for both onnxruntime and onnxruntime-gpu

Original file line number	Diff line number	Diff line change
`@@ -109,6 +109,7 @@ def _is_package_available(pkg_name: str):`
`109`	`109`	`"onnxruntime-rocm",`
`110`	`110`	`"onnxruntime-migraphx",`
`111`	`111`	`"onnxruntime-training",`
	`112`	`+ "onnxruntime-vitisai",`
`112`	`113`	`)`
`113`	`114`	`_onnxruntime_version = None`
`114`	`115`	`# For the metadata, we have to look for both onnxruntime and onnxruntime-gpu`