From 8397accfc4592b9901f6df6782c762957b94bda0 Mon Sep 17 00:00:00 2001
From: lakshay sharma <31830611+Lakshaysharma048@users.noreply.github.com>
Date: Fri, 20 Dec 2024 13:09:17 -0800
Subject: [PATCH] Update import_utils.py

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 3014efebc82e..ad2619b569b3 100644
--- a/src/diffusers/utils/import_utils.py
+++ b/src/diffusers/utils/import_utils.py
@@ -143,6 +143,7 @@
         "ort_nightly_directml",
         "onnxruntime-rocm",
         "onnxruntime-training",
+        "onnxruntime-vitisai",
     )
     _onnxruntime_version = None
     # For the metadata, we have to look for both onnxruntime and onnxruntime-gpu