From c6f1bc65c0c55de131820aa194ec391a9229a0eb Mon Sep 17 00:00:00 2001 From: hzhaoy Date: Thu, 4 Jul 2024 10:20:28 +0800 Subject: [PATCH] tiny fix Former-commit-id: 8f43ad988a4fd518a708fba53a173596ce2c59dd --- src/llamafactory/data/collator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llamafactory/data/collator.py b/src/llamafactory/data/collator.py index 91871eaa..c878dcd3 100644 --- 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], ] ] ]