From d31df04c8aa84da61f96862a3967d2a9900bb9b0 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sat, 17 Aug 2024 23:00:44 -0400 Subject: [PATCH] Indentation. --- comfy/ldm/flux/model.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py index eb7767c..63970ca 100644 --- a/comfy/ldm/flux/model.py +++ b/comfy/ldm/flux/model.py @@ -115,14 +115,14 @@ class Flux(nn.Module): pe = self.pe_embedder(ids) for i, block in enumerate(self.double_blocks): - img, txt = block(img=img, txt=txt, vec=vec, pe=pe) - - if control is not None: # Controlnet - control_i = control.get("input") - if i < len(control_i): - add = control_i[i] - if add is not None: - img += add + img, txt = block(img=img, txt=txt, vec=vec, pe=pe) + + if control is not None: # Controlnet + control_i = control.get("input") + if i < len(control_i): + add = control_i[i] + if add is not None: + img += add img = torch.cat((txt, img), 1)