423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2026-03-10 22:06:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4673 from hzhaoy/main

Browse Source 
tiny fix

Former-commit-id: 07d96d497c
This commit is contained in:
hoshi-hiyouga
2024-07-04 10:40:41 +08:00
committed by GitHub
parent aa14a625e4 6d892dbc23
commit d08456c0ce
1 changed files with 1 additions and 1 deletions
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],
] ]
] ]
] ]