From 7e50fa1b2b96a847a9ce65d19f6eddffddf5828c Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Sat, 30 Mar 2024 16:44:00 +0900 Subject: [PATCH] fix onnx model loading --- modules/onnx_impl/pipelines/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules/onnx_impl/pipelines/__init__.py b/modules/onnx_impl/pipelines/__init__.py index 3d2f217df..882eb3c48 100644 --- a/modules/onnx_impl/pipelines/__init__.py +++ b/modules/onnx_impl/pipelines/__init__.py @@ -64,6 +64,10 @@ class PipelineBase(TorchCompatibleModule, diffusers.DiffusionPipeline, metaclass log.debug(f"Component device/dtype conversion failed: module={name} args={args}, kwargs={kwargs}") return self + @property + def components(self): + return {} + @classmethod def from_pretrained(cls, pretrained_model_name_or_path, **_): # pylint: disable=arguments-differ return OnnxRawPipeline(