From 1d7ae838377145deef780cad62dfe4fcdaefb70b Mon Sep 17 00:00:00 2001 From: H1yori233 Date: Sat, 28 Feb 2026 02:45:54 -0800 Subject: [PATCH] precommit --- fastvideo/models/dits/matrixgame/causal_model.py | 1 - .../preprocess/matrixgame/matrixgame_preprocess_pipeline.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/fastvideo/models/dits/matrixgame/causal_model.py b/fastvideo/models/dits/matrixgame/causal_model.py index b3360e243..99bb8e134 100644 --- a/fastvideo/models/dits/matrixgame/causal_model.py +++ b/fastvideo/models/dits/matrixgame/causal_model.py @@ -487,7 +487,6 @@ class CausalMatrixGameTransformerBlock(nn.Module): assert temb.ndim == 4 num_frames = temb.shape[1] - frame_seqlen = hidden_states.shape[1] // num_frames bs, seq_length, _ = hidden_states.shape orig_dtype = hidden_states.dtype diff --git a/fastvideo/pipelines/preprocess/matrixgame/matrixgame_preprocess_pipeline.py b/fastvideo/pipelines/preprocess/matrixgame/matrixgame_preprocess_pipeline.py index c762d03d3..f02fc8e00 100644 --- a/fastvideo/pipelines/preprocess/matrixgame/matrixgame_preprocess_pipeline.py +++ b/fastvideo/pipelines/preprocess/matrixgame/matrixgame_preprocess_pipeline.py @@ -159,8 +159,8 @@ class PreprocessPipeline_MatrixGame(BasePreprocessPipeline): assert kb_length >= num_frames, ( f"keyboard length {kb_length} is smaller than " f"num_frames {num_frames} for {action_path}.") - keyboard_cond_list.append(action_data.astype( - np.float32)[:num_frames]) + keyboard_cond_list.append( + action_data.astype(np.float32)[:num_frames]) if keyboard_cond_list: features["keyboard_cond"] = keyboard_cond_list if mouse_cond_list: