Update import_utils.py (#10329)

added onnxruntime-vitisai for custom build onnxruntime pkg
2026-06-02 00:01:34 +08:00 · 2025-04-02 12:47:10 -07:00
parent 78c2fdc52e
commit b0ff822ed3
1 changed files with 1 additions and 0 deletions
--- a/src/diffusers/utils/import_utils.py
+++ b/src/diffusers/utils/import_utils.py
@@ -109,6 +109,7 @@ if _onnx_available:
        "onnxruntime-rocm",
        "onnxruntime-migraphx",
        "onnxruntime-training",
+        "onnxruntime-vitisai",
    )
    _onnxruntime_version = None
    # For the metadata, we have to look for both onnxruntime and onnxruntime-gpu