From 5a6258d11ec8eb7bf1d458883f132370527251d4 Mon Sep 17 00:00:00 2001 From: Rachit Shah Date: Sat, 5 Oct 2024 01:33:42 +0530 Subject: [PATCH] config attribute not foud error for FluxImagetoImage Pipeline for multi controlnet solved --- .../flux/pipeline_flux_controlnet_image_to_image.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py index deeb9e3f546a..b9c040d3e30b 100644 --- a/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py +++ b/src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py @@ -872,9 +872,12 @@ def __call__( timestep = t.expand(latents.shape[0]).to(latents.dtype) - guidance = ( - torch.tensor([guidance_scale], device=device) if self.controlnet.config.guidance_embeds else None - ) + if isinstance(self.controlnet, FluxMultiControlNetModel): + use_guidance = self.controlnet.nets[0].config.guidance_embeds + else: + use_guidance = self.controlnet.config.guidance_embeds + + guidance = torch.tensor([guidance_scale], device=device) if use_guidance else None guidance = guidance.expand(latents.shape[0]) if guidance is not None else None controlnet_block_samples, controlnet_single_block_samples = self.controlnet(