lInTiNG

pytorch · Aug 5, 2024 · 4e6be43 · 4e6be43
1 parent e76304c
commit 4e6be43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchtune/utils/collate.py b/torchtune/utils/collate.py
@@ -68,4 +68,4 @@ def padded_collate(
             (0, labels_seq_len - input_ids_seq_len),
             value=padding_idx,
         )
-    return {"tokens": input_ids.long(), "labels": labels.long()}
+    return {"tokens": input_ids.long(), "labels": labels.long()}