From 921e577ad9c997629bc01c7bc5137ef6d08a3764 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 26 Mar 2025 18:59:13 +0200
Subject: [PATCH] Update model.py

---
 wanvideo/modules/model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index c76513b..f9ff249 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -555,8 +555,7 @@ class WanAttentionBlock(nn.Module):
                 return x
                 
             else:
-                cross_attn_result = self.cross_attn(self.norm3(x), context, context_lens, clip_embed=clip_embed)
-                x = x + cross_attn_result
+                x = x + self.cross_attn(self.norm3(x), context, context_lens, clip_embed=clip_embed)
                 y = self.ffn(self.norm2(x).float() * (1 + e[4]) + e[3])
                 x = x.to(torch.float32) + (y.to(torch.float32) * e[5].to(torch.float32))
                 return x