Skip to content

GitLab

Unverified Commit b0ff822e authored by lakshay sharma's avatar lakshay sharma Committed by GitHub
Browse files

Update import_utils.py (#10329) 

added onnxruntime-vitisai for custom build onnxruntime pkg
parent 78c2fdc5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/utils/import_utils.py
View file @ b0ff822e
...@@ -109,6 +109,7 @@ if _onnx_available: ...@@ -109,6 +109,7 @@ if _onnx_available:
"onnxruntime-rocm", "onnxruntime-rocm",
"onnxruntime-migraphx", "onnxruntime-migraphx",
"onnxruntime-training", "onnxruntime-training",
"onnxruntime-vitisai",
) )
_onnxruntime_version = None _onnxruntime_version = None
# For the metadata, we have to look for both onnxruntime and onnxruntime-gpu # For the metadata, we have to look for both onnxruntime and onnxruntime-gpu
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment