remove unnecessary assert

chengzeyi · Dec 19, 2024 · 7d11f6c · 7d11f6c
1 parent a9defb2
commit 7d11f6c
Showing 1 changed file with 0 additions and 2 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
@@ -146,8 +146,6 @@ def new_transformer_block_forward(
             ):
                 world_size = DP.get_world_size(seq_mesh)
                 if attention_mask is not None:
-                    assert attention_mask.shape[0] == 1, "Only support batch size 1 for now"
-
                     hidden_states_len = hidden_states.shape[-2]
                     encoder_hidden_states_len = encoder_hidden_states.shape[-2]