add extra test for pixtral mm_input
This commit is contained in:
1 parent
8f3bff456b
commit
0fc949783d
2 files changed
+40
-4
No files matched your search
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user