From e7c39757f6f7b365850fad5a16914f16c96605dc Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 12 Jun 2025 19:04:15 +0300
Subject: [PATCH] Allow compiling crossattn

I don't remember why I even disabled this... seems to work, big effect with NAG
---
 wanvideo/modules/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 11ed685..e3b3753 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -643,7 +643,7 @@ class WanAttentionBlock(nn.Module):
                                     num_latent_frames=num_latent_frames, nag_params=nag_params, nag_context=nag_context)
         del e
         return x
-    @torch.compiler.disable()
+    #@torch.compiler.disable()
     def cross_attn_ffn(self, x, context, context_lens, e, clip_embed=None, grid_sizes=None, 
                        audio_proj=None, audio_context_lens=None, audio_scale=1.0, num_latent_frames=21, nag_params={}, nag_context=None):
             x = x + self.cross_attn(self.norm3(x), context, context_lens, clip_embed=clip_embed,