Unverified Commit 24c7d578 authored by Rachit Shah's avatar Rachit Shah Committed by GitHub
Browse files

config attribute not foud error for FluxImagetoImage Pipeline for multi controlnet solved (#9586)


Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent bfa0aa4f
...@@ -903,9 +903,12 @@ class FluxControlNetImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From ...@@ -903,9 +903,12 @@ class FluxControlNetImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin, From
timestep = t.expand(latents.shape[0]).to(latents.dtype) timestep = t.expand(latents.shape[0]).to(latents.dtype)
guidance = ( if isinstance(self.controlnet, FluxMultiControlNetModel):
torch.tensor([guidance_scale], device=device) if self.controlnet.config.guidance_embeds else None 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 guidance = guidance.expand(latents.shape[0]) if guidance is not None else None
if isinstance(controlnet_keep[i], list): if isinstance(controlnet_keep[i], list):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment