modify style & little change

src/llamafactory/data/collator.py

@@ -99,6 +99,9 @@ class MultiModalDataCollatorForSeq2Seq(DataCollatorForSeq2Seq):
 
         features: Dict[str, "torch.Tensor"] = super().__call__(features)
         features.update(mm_inputs)
+        if features.get("pixel_values") is not None and isinstance(features["pixel_values"], list):
+            features = features.data
+
         return features