Skip to content

Commit 0b7225e

Browse files
authored
Add ort_nightly_directml to the onnxruntime candidates (huggingface#1458)
* Add `ort_nightly_directml` to the `onnxruntime` candidates * style
1 parent db7b7bd commit 0b7225e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/diffusers/utils/import_utils.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,13 @@
157157
_onnxruntime_version = "N/A"
158158
_onnx_available = importlib.util.find_spec("onnxruntime") is not None
159159
if _onnx_available:
160-
candidates = ("onnxruntime", "onnxruntime-gpu", "onnxruntime-directml", "onnxruntime-openvino")
160+
candidates = (
161+
"onnxruntime",
162+
"onnxruntime-gpu",
163+
"onnxruntime-directml",
164+
"onnxruntime-openvino",
165+
"ort_nightly_directml",
166+
)
161167
_onnxruntime_version = None
162168
# For the metadata, we have to look for both onnxruntime and onnxruntime-gpu
163169
for pkg in candidates:

0 commit comments

Comments
 (0)