Skip to content

Commit c4892f1

Browse files
authored
Update onnxruntime package candidates (#2666)
* update import onnxruntime package, enable onnxruntime-rocm and onnxruntime-training * add ort_nightly_gpu
1 parent f6feb69 commit c4892f1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/utils/import_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,9 +153,12 @@
153153
candidates = (
154154
"onnxruntime",
155155
"onnxruntime-gpu",
156+
"ort_nightly_gpu",
156157
"onnxruntime-directml",
157158
"onnxruntime-openvino",
158159
"ort_nightly_directml",
160+
"onnxruntime-rocm",
161+
"onnxruntime-training",
159162
)
160163
_onnxruntime_version = None
161164
# For the metadata, we have to look for both onnxruntime and onnxruntime-gpu

0 commit comments

Comments
 (0)