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

Fix "ort is not defined" issue.
dereyly · Sep 23, 2022 · 1e15203 · 1e15203
1 parent 8211b62
commit 1e15203
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/onnx_utils.py b/src/diffusers/onnx_utils.py
@@ -114,7 +114,7 @@ def _from_pretrained(
         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,
     ):
         """