Skip to content

GitLab

Unverified Commit 3724d5f6 authored by Chris's avatar Chris Committed by GitHub
Browse files

[Bugfix][Model] Fix Python 3.8 compatibility in Pixtral model by updating type annotations (#8490)

parent 50e9ec41
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
vllm/model_executor/models/pixtral.py
View file @ 3724d5f6
...@@ -454,7 +454,7 @@ class Transformer(nn.Module): ...@@ -454,7 +454,7 @@ class Transformer(nn.Module):
return x return x
def position_meshgrid(patch_embeds_list: list[torch.Tensor], ) -> torch.Tensor: def position_meshgrid(patch_embeds_list: List[torch.Tensor], ) -> torch.Tensor:
positions = torch.cat([ positions = torch.cat([
torch.stack( torch.stack(
torch.meshgrid( torch.meshgrid(
......
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