Merge branch 'main' of https://github.com/hiyouga/LLaMA-Factory

Former-commit-id: f0b54254b43e93063232f633cdcf1e31d1419bfe
2024-07-04 14:23:37 +08:00
parent 6685f1fb9e c89ff328f6
commit a3e4f2b716
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],
            ]
        ]
    ]