diff --git a/src/diffusers/modular_pipelines/ernie_image/modular_pipeline.py b/src/diffusers/modular_pipelines/ernie_image/modular_pipeline.py index cf4497fe9138..f4cb2204369c 100644 --- a/src/diffusers/modular_pipelines/ernie_image/modular_pipeline.py +++ b/src/diffusers/modular_pipelines/ernie_image/modular_pipeline.py @@ -15,6 +15,7 @@ import torch from ...configuration_utils import ConfigMixin, register_to_config +from ...loaders import ErnieImageLoraLoaderMixin from ...utils import logging from ..modular_pipeline import ModularPipeline @@ -63,7 +64,7 @@ def unpack_latents(self, latents: torch.Tensor) -> torch.Tensor: ) -class ErnieImageModularPipeline(ModularPipeline): +class ErnieImageModularPipeline(ModularPipeline, ErnieImageLoraLoaderMixin): """ A ModularPipeline for ErnieImage.