From 1e152030bd836e0b2bfc5fabe2186325b8d38a04 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Fri, 23 Sep 2022 17:02:03 +0200 Subject: [PATCH] Fix breaking error: "ort is not defined" (#626) Fix "ort is not defined" issue. --- src/diffusers/onnx_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/onnx_utils.py b/src/diffusers/onnx_utils.py index 9320ca9b..2282f411 100644 --- a/src/diffusers/onnx_utils.py +++ b/src/diffusers/onnx_utils.py @@ -114,7 +114,7 @@ class OnnxRuntimeModel: cache_dir: Optional[str] = None, file_name: Optional[str] = None, provider: Optional[str] = None, - sess_options: Optional[ort.SessionOptions] = None, + sess_options: Optional["ort.SessionOptions"] = None, **kwargs, ): """