Skip to content

Commit 5a6258d

Browse files
committed
config attribute not foud error for FluxImagetoImage Pipeline for multi controlnet solved
1 parent 99f6082 commit 5a6258d

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/diffusers/pipelines/flux/pipeline_flux_controlnet_image_to_image.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -872,9 +872,12 @@ def __call__(
872872

873873
timestep = t.expand(latents.shape[0]).to(latents.dtype)
874874

875-
guidance = (
876-
torch.tensor([guidance_scale], device=device) if self.controlnet.config.guidance_embeds else None
877-
)
875+
if isinstance(self.controlnet, FluxMultiControlNetModel):
876+
use_guidance = self.controlnet.nets[0].config.guidance_embeds
877+
else:
878+
use_guidance = self.controlnet.config.guidance_embeds
879+
880+
guidance = torch.tensor([guidance_scale], device=device) if use_guidance else None
878881
guidance = guidance.expand(latents.shape[0]) if guidance is not None else None
879882

880883
controlnet_block_samples, controlnet_single_block_samples = self.controlnet(

0 commit comments

Comments
 (0)