Order of operations matters.

comfyanonymous · Aug 13, 2024 · 5942c17 · 5942c17
1 parent c032b11
commit 5942c17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comfy/ldm/flux/layers.py b/comfy/ldm/flux/layers.py
@@ -170,8 +170,8 @@ def forward(self, img: Tensor, txt: Tensor, vec: Tensor, pe: Tensor):
         txt_attn, img_attn = attn[:, : txt.shape[1]], attn[:, txt.shape[1] :]
 
         # calculate the img bloks
-        img += img_mod1.gate * self.img_attn.proj(img_attn)
-        img += img_mod2.gate * self.img_mlp((1 + img_mod2.scale) * self.img_norm2(img) + img_mod2.shift)
+        img = img + img_mod1.gate * self.img_attn.proj(img_attn)
+        img = img + img_mod2.gate * self.img_mlp((1 + img_mod2.scale) * self.img_norm2(img) + img_mod2.shift)
 
         # calculate the txt bloks
         txt += txt_mod1.gate * self.txt_attn.proj(txt_attn)