From b0ff822ed385a2bedf52c57472766f9f95e83002 Mon Sep 17 00:00:00 2001
From: lakshay sharma <31830611+Lakshaysharma048@users.noreply.github.com>
Date: Wed, 2 Apr 2025 12:47:10 -0700
Subject: [PATCH] Update import_utils.py (#10329)

added onnxruntime-vitisai for custom build onnxruntime pkg
---
 src/diffusers/utils/import_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py
index 98b9c75451..f61116aaaf 100644
--- 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