Converting sample labels to list

Signed-off-by: romitjain <[email protected]>

Author: romitjain

plugins/online-data-mixing/src/fms_acceleration_odm/odm/dataset.py

@@ -243,7 +243,7 @@ def __next__(self):
                     else torch.ones_like(sample["input_ids"][0])
                 ),
                 "labels": (
-                    sample["labels"][0]
+                    sample["labels"][0].tolist()
                     if "labels" in sample
                     else sample["input_ids"][0]
                 ),