make hunyuan_video roboster

chengzeyi · Dec 25, 2024 · fbb42ff · fbb42ff
1 parent 1c83112
commit fbb42ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/para_attn/context_parallel/diffusers_adapters/hunyuan_video.py b/src/para_attn/context_parallel/diffusers_adapters/hunyuan_video.py
@@ -29,6 +29,7 @@ def new_forward(
         guidance: torch.Tensor = None,
         attention_kwargs: Optional[Dict[str, Any]] = None,
         return_dict: bool = True,
+        **kwargs,
     ) -> Union[torch.Tensor, Dict[str, torch.Tensor]]:
         if attention_kwargs is not None:
             attention_kwargs = attention_kwargs.copy()

diff --git a/src/para_attn/focus_attn/diffusers_adapters/hunyuan_video.py b/src/para_attn/focus_attn/diffusers_adapters/hunyuan_video.py
@@ -30,6 +30,7 @@ def new_forward(
         guidance: torch.Tensor = None,
         attention_kwargs: Optional[Dict[str, Any]] = None,
         return_dict: bool = True,
+        **kwargs,
     ) -> Union[torch.Tensor, Dict[str, torch.Tensor]]:
         if attention_kwargs is not None:
             attention_kwargs = attention_kwargs.copy()