Skip to content

Commit 1a1eba3

Browse files
authored
Update src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py
1 parent 9915ebe commit 1a1eba3

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/diffusers/pipelines/flux/pipeline_flux_controlnet_inpainting.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1097,11 +1097,13 @@ def __call__(
10971097
# predict the noise residual
10981098
if isinstance(self.controlnet, FluxMultiControlNetModel):
10991099
use_guidance = self.controlnet.nets[0].config.guidance_embeds
1100-
guidance = torch.tensor([guidance_scale], device=device) if use_guidance else None
1101-
else :
1100+
else:
11021101
use_guidance = self.controlnet.config.guidance_embeds
1103-
guidance = torch.full([1], guidance_scale, device=device, dtype=torch.float32) if use_guidance else None
1104-
guidance = guidance.expand(latents.shape[0]) if use_guidance else None
1102+
if use_guidance:
1103+
guidance = torch.full([1], guidance_scale, device=device, dtype=torch.float32)
1104+
guidance = guidance.expand(latents.shape[0])
1105+
else:
1106+
guidance = None
11051107

11061108
if isinstance(controlnet_keep[i], list):
11071109
cond_scale = [c * s for c, s in zip(controlnet_conditioning_scale, controlnet_keep[i])]

0 commit comments

Comments
 (0)