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, ): """