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