fix style

Former-commit-id: 76a36d9acecbf36b6959a14caacfed1d32bcee41
2025-01-13 14:19:38 +08:00
parent 07798e4aad
commit 4b61610b12
3 changed files with 23 additions and 17 deletions
--- a/src/llamafactory/data/collator.py
+++ b/src/llamafactory/data/collator.py
@@ -157,7 +157,7 @@ class MultiModalDataCollatorForSeq2Seq(DataCollatorForSeq2Seq):
            features["position_ids"] = [torch.arange(input_ids.size(0)).long() for input_ids in features["input_ids"]]
            features["position_ids"] = pad_sequence(features["position_ids"], batch_first=True, padding_value=0)
            new_features = {"data": features}
-            new_features.update({"labels": features['labels']})
+            new_features.update({"labels": features["labels"]})
            features = new_features

        return features