From 55db8f390909d7aa063f220d221ebef93f389d2d Mon Sep 17 00:00:00 2001 From: Samuel Tallet Date: Sun, 14 Jun 2026 14:01:33 +0200 Subject: [PATCH] Enable LoRA loading on ErnieImageModularPipeline through load_lora_weights --- .../modular_pipelines/ernie_image/modular_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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.