"vscode:/vscode.git/clone" did not exist on "b36adfa349cfab0e79f3d736d5e5413bd3ee19f5"
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):
pixel_values: Annotated[
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)]
......
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