Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

check for init and call

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-04-03 19:23:33 -04:00
parent 8725cfc488
commit ea0389162f
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/onnx_impl/pipelines/__init__.py
+1 -1
View File
@@ -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
modules/sd_models_utils.py
+3 -1
View File
@@ -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