diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index 37535366ed44..9f16e0256c25 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -151,6 +151,7 @@ "onnxruntime-rocm", "onnxruntime-migraphx", "onnxruntime-training", + "onnxruntime-vitisai", ) _onnxruntime_version = None # For the metadata, we have to look for both onnxruntime and onnxruntime-gpu