p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
hiyouga 2024-07-04 14:23:37 +08:00
parent 1e27e8c776 07d96d497c
commit 226a9e563f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/data/collator.py
View File

@ -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, x, x, x],
[x, x, o, o, x, x], [x, x, o, o, x, x],
[x, x, o, o, o, x], [x, x, o, o, o, x],
[x, x, o, x, x, x], [x, x, x, x, x, x],
] ]
] ]
] ]