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