From 8479624614ec0d52e982bbbab633736fb1a15eef Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 20 Jun 2025 17:31:01 +0300
Subject: [PATCH] fix split prompting

---
 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 e1ba09d..d7d2a49 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -705,7 +705,7 @@ class WanAttentionBlock(nn.Module):
             x_segment = x[:, segment_indices, :]
             
             # Process segment with its prompt and clip embedding
-            processed_segment = self.cross_attn(self.norm3(x_segment), segment_context, segment_context_lens, clip_embed=segment_clip_embed, nag_scale=nag_scale)
+            processed_segment = self.cross_attn(self.norm3(x_segment), segment_context, segment_context_lens, clip_embed=segment_clip_embed)
             processed_segment = processed_segment.to(x.dtype)
             
             # Add to combined result