tiny fix

Former-commit-id: 8f43ad988a4fd518a708fba53a173596ce2c59dd
2025-10-15 08:08:09 +08:00 · 2024-07-04 10:20:28 +08:00 · 2024-07-04 10:20:28 +08:00 · c6f1bc65c0
commit c6f1bc65c0
parent 0f43c61229
1 changed files with 1 additions and 1 deletions
--- a/src/llamafactory/data/collator.py
+++ b/src/llamafactory/data/collator.py
@ -41,7 +41,7 @@ def prepare_4d_attention_mask(attention_mask_with_indices: "torch.Tensor", dtype
                [x, x, o, x, x, x],
                [x, x, o, o, x, x],
                [x, x, o, o, o, x],
-                [x, x, o, x, x, x],
+                [x, x, x, x, x, x],
            ]
        ]
    ]