tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

stack

Browse Source
This commit is contained in:
Sunny Liu
2025-02-04 21:25:13 -05:00
parent adcbc7459b
commit 3f6be519d5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/monkeypatch/utils.py
View File

@@ -137,7 +137,7 @@ def get_packed_mask_from_pos_ids(position_ids):
results.append(doc_mask) results.append(doc_mask)
return results return torch.stack(results)
def get_seqlens_from_pos_ids(position_ids): def get_seqlens_from_pos_ids(position_ids):