From 6fb7b0a05e3758e41e3871804598cea18a0b42fb Mon Sep 17 00:00:00 2001 From: hyw <109567717+yuweih205@users.noreply.github.com> Date: Wed, 9 Sep 2026 22:34:46 +0800 Subject: [PATCH] [Model] Skip unused frame packing in Wan2.2 S2V (#7155) Signed-off-by: hyw --- .../diffusion/models/wan2_2/wan2_2_s2v_transformer.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/vllm_omni/diffusion/models/wan2_2/wan2_2_s2v_transformer.py b/vllm_omni/diffusion/models/wan2_2/wan2_2_s2v_transformer.py index d1b74d020..cce5e43be 100644 --- a/vllm_omni/diffusion/models/wan2_2/wan2_2_s2v_transformer.py +++ b/vllm_omni/diffusion/models/wan2_2/wan2_2_s2v_transformer.py @@ -1,5 +1,5 @@ # SPDX-License-Identifier: Apache-2.0 -# SPDX-FileCopyrightText: Copyright contributors to the vLLM project +# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project """ Wan2.2 Speech-to-Video (S2V) Transformer using vllm-omni ops. @@ -1359,11 +1359,9 @@ class WanS2VTransformer3DModel(nn.Module): return flatten_mot, mot_remb def process_motion_frame_pack(self, motion_latents, drop_motion_frames=False, add_last_motion=2): - flatten_mot, mot_remb = self.frame_packer(motion_latents, add_last_motion) if drop_motion_frames: - return [m[:, :0] for m in flatten_mot], [m[:, :0] for m in mot_remb] - else: - return flatten_mot, mot_remb + return [], [] + return self.frame_packer(motion_latents, add_last_motion) def process_motion_transformer_motioner(self, motion_latents, drop_motion_frames=False, add_last_motion=True): batch_size = motion_latents.shape[0]