fix system prompt and tests

Former-commit-id: 955efca677b299749f3d40d587ee310951537543
2025-01-13 14:18:06 +08:00
parent 6d6acd0213
commit 07798e4aad
4 changed files with 17 additions and 10 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(features)
+            new_features.update({"labels": features['labels']})
            features = new_features

        return features