add extra test for pixtral mm_input

Former-commit-id: c706ec8a5dbd3c72ab15a709668624c0c7bbd8ce
2024-10-15 17:09:24 +08:00
parent c9e77ab352
commit f463b3f038
2 changed files with 40 additions and 4 deletions
--- a/src/llamafactory/data/mm_plugin.py
+++ b/src/llamafactory/data/mm_plugin.py
@@ -513,8 +513,6 @@ class PixtralPlugin(BasePlugin):
    ) -> Dict[str, Union[List[int], "torch.Tensor"]]:
        self._validate_input(images, videos)
        mm_inputs = self._get_mm_inputs(images, videos, processor)
-        if mm_inputs.get("image_sizes"):
-            mm_inputs.pop("image_sizes")

        return mm_inputs