From 7e2810b853db3473e4d087aa98b61d0a991077b6 Mon Sep 17 00:00:00 2001 From: fsx950223 Date: Thu, 16 Mar 2023 12:10:05 +0800 Subject: [PATCH] add rocm onnxruntime support --- 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 b3c6d1824369..7f1cdfa02784 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -153,6 +153,7 @@ candidates = ( "onnxruntime", "onnxruntime-gpu", + "onnxruntime-rocm", "onnxruntime-directml", "onnxruntime-openvino", "ort_nightly_directml",