stack

2025-02-04 21:25:13 -05:00
parent adcbc7459b
commit 3f6be519d5
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/monkeypatch/utils.py
+++ b/src/axolotl/monkeypatch/utils.py
@@ -137,7 +137,7 @@ def get_packed_mask_from_pos_ids(position_ids):

        results.append(doc_mask)

-    return results
+    return torch.stack(results)


 def get_seqlens_from_pos_ids(position_ids):