diff --git a/nodes/ParaAttentionNode.py b/nodes/ParaAttentionNode.py
index 86a939c..8d7cd6b 100644
--- a/nodes/ParaAttentionNode.py
+++ b/nodes/ParaAttentionNode.py
@@ -129,12 +129,12 @@ def fb_cache_patch_dit_exit(img, transformer_options):
     return img
 
 def fb_cache_prepare_wrapper(wrapper_executor, noise, latent_image, sampler, sigmas, denoise_mask=None,
-                                  callback=None, disable_pbar=False, seed=None):
+                                  callback=None, disable_pbar=False, seed=None, **kwargs):
     cfg_guider = wrapper_executor.class_obj
 
     try:
         out = wrapper_executor(noise, latent_image, sampler, sigmas, denoise_mask=denoise_mask, callback=callback,
-                               disable_pbar=disable_pbar, seed=seed)
+                               disable_pbar=disable_pbar, seed=seed, **kwargs)
     finally:
         diffusion_model = cfg_guider.model_patcher.model.diffusion_model
         if hasattr(diffusion_model, fb_cache_model_temp):
diff --git a/nodes/TeaCacheNode.py b/nodes/TeaCacheNode.py
index 82e1f05..8355433 100644
--- a/nodes/TeaCacheNode.py
+++ b/nodes/TeaCacheNode.py
@@ -225,7 +225,7 @@ def tea_cache_patch_dit_exit(img, transformer_options):
     return img
 
 def tea_cache_prepare_wrapper(wrapper_executor, noise, latent_image, sampler, sigmas, denoise_mask=None,
-                                  callback=None, disable_pbar=False, seed=None):
+                                  callback=None, disable_pbar=False, seed=None, **kwargs):
     cfg_guider = wrapper_executor.class_obj
 
     # Use cfd_guider.model_options, which is copied from modelPatcher.model_options and will be restored after execution without any unexpected contamination
@@ -234,7 +234,7 @@ def tea_cache_prepare_wrapper(wrapper_executor, noise, latent_image, sampler, si
     temp_options['cnt'] = 0
     try:
         out = wrapper_executor(noise, latent_image, sampler, sigmas, denoise_mask=denoise_mask, callback=callback,
-                               disable_pbar=disable_pbar, seed=seed)
+                               disable_pbar=disable_pbar, seed=seed, **kwargs)
     finally:
         diffusion_model = cfg_guider.model_patcher.model.diffusion_model
         if hasattr(diffusion_model, "flux_tea_cache"):