Fix breaking error: "ort is not defined" (#626)

Fix "ort is not defined" issue.

Fix breaking error: "ort is not defined" (#626)
Fix "ort is not defined" issue.
1e152030 · Pedro Cuenca · GitHub · 8211b622 · 1e152030
Unverified Commit 1e152030 authored Sep 23, 2022 by Pedro Cuenca Committed by GitHub Sep 23, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/onnx_utils.py src/diffusers/onnx_utils.py +1 -1

No files found.
--- 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,
    ):
        """