From 515d3edeb60087673b28b23f988af798b0b2de63 Mon Sep 17 00:00:00 2001 From: peixuanzuo Date: Tue, 14 Mar 2023 09:27:31 +0000 Subject: [PATCH 1/2] update import onnxruntime package, enable onnxruntime-rocm and onnxruntime-training --- src/diffusers/utils/import_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index b3c6d1824369..976e64af0ad8 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -156,6 +156,8 @@ "onnxruntime-directml", "onnxruntime-openvino", "ort_nightly_directml", + "onnxruntime-rocm", + "onnxruntime-training", ) _onnxruntime_version = None # For the metadata, we have to look for both onnxruntime and onnxruntime-gpu From 742dc6f1a7b169e7c8580e7fb060320b12ce3450 Mon Sep 17 00:00:00 2001 From: peixuanzuo Date: Fri, 17 Mar 2023 05:59:28 +0000 Subject: [PATCH 2/2] add ort_nightly_gpu --- src/diffusers/utils/import_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index 976e64af0ad8..935312136d02 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -153,6 +153,7 @@ candidates = ( "onnxruntime", "onnxruntime-gpu", + "ort_nightly_gpu", "onnxruntime-directml", "onnxruntime-openvino", "ort_nightly_directml",