Unverified Commit ebafb093 authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Bugfix] Allow dynamic number of patches for llava_onevision (#23525)


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent 0cb7b065
...@@ -75,7 +75,7 @@ class LlavaOnevisionImagePixelInputs(TensorSchema): ...@@ -75,7 +75,7 @@ class LlavaOnevisionImagePixelInputs(TensorSchema):
pixel_values: Annotated[ pixel_values: Annotated[
Union[torch.Tensor, list[torch.Tensor]], Union[torch.Tensor, list[torch.Tensor]],
TensorShape("bn", "np", 3, "h", "w"), TensorShape("bn", "np", 3, "h", "w", dynamic_dims={"np"}),
] ]
image_sizes: Annotated[Optional[torch.Tensor], TensorShape("bn", 2)] image_sizes: Annotated[Optional[torch.Tensor], TensorShape("bn", 2)]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment