diff --git a/diffsynth_engine/models/qwen_image/qwen_image_dit_nunchaku.py b/diffsynth_engine/models/qwen_image/qwen_image_dit_nunchaku.py
index 3c580ce..e583aae 100644
--- a/diffsynth_engine/models/qwen_image/qwen_image_dit_nunchaku.py
+++ b/diffsynth_engine/models/qwen_image/qwen_image_dit_nunchaku.py
@@ -179,6 +179,7 @@ def forward(
         rotary_emb: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,
         attn_mask: Optional[torch.Tensor] = None,
         attn_kwargs: Optional[Dict[str, Any]] = None,
+        modulate_index: Optional[List[int]] = None,
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         if self.use_nunchaku_awq:
             img_mod_params = self.img_mod(temb)  # [B, 6*dim]