From 738df477485de3633049651a9f1d498adf95a3d5 Mon Sep 17 00:00:00 2001 From: hzhaoy Date: Thu, 4 Jul 2024 10:20:28 +0800 Subject: [PATCH] tiny fix --- 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], ] ] ]