fix onnx model loading

This commit is contained in:
Seunghoon Lee
2024-03-30 16:44:00 +09:00
parent bd7ba768b1
commit 7e50fa1b2b
+4
View File
@@ -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(