precommit

2026-02-28 02:45:54 -08:00
parent b5644a1faf
commit 1d7ae83837
2 changed files with 2 additions and 3 deletions
@@ -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
@@ -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: