From ea0389162f6fa8de7cb0859fb2c1661c41b37fba Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 3 Apr 2025 19:23:33 -0400 Subject: [PATCH] check for init and call Signed-off-by: Vladimir Mandic --- modules/onnx_impl/pipelines/__init__.py | 2 +- modules/sd_models_utils.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/onnx_impl/pipelines/__init__.py b/modules/onnx_impl/pipelines/__init__.py index 62db80cd5..6d035771d 100644 --- a/modules/onnx_impl/pipelines/__init__.py +++ b/modules/onnx_impl/pipelines/__init__.py @@ -47,7 +47,7 @@ class PipelineBase(TorchCompatibleModule, diffusers.DiffusionPipeline, metaclass if "optimum.onnxruntime" in sys.modules: import optimum.onnxruntime - if isinstance(module, optimum.onnxruntime.modeling_diffusion._ORTDiffusionModelPart): # pylint: disable=protected-access + if isinstance(module, optimum.onnxruntime.modeling_diffusion._ORTDiffusionModelPart): # pylint: disable=protected-access, no-member device = extract_device(args, kwargs) if device is None: return self diff --git a/modules/sd_models_utils.py b/modules/sd_models_utils.py index 0b2794bac..ae50e5748 100644 --- a/modules/sd_models_utils.py +++ b/modules/sd_models_utils.py @@ -19,12 +19,14 @@ class NoWatermark: def get_signature(cls): + if cls is None or not hasattr(cls, '__init__'): + return {} signature = inspect.signature(cls.__init__, follow_wrapped=True) return signature.parameters def get_call(cls): - if cls is None: + if cls is None or not hasattr(cls, '__call__'): # noqa: B004 return [] signature = inspect.signature(cls.__call__, follow_wrapped=True) return signature.parameters