diff --git a/vllm/model_executor/models/pixtral.py b/vllm/model_executor/models/pixtral.py index b26fd558..682b78bb 100644 --- a/vllm/model_executor/models/pixtral.py +++ b/vllm/model_executor/models/pixtral.py @@ -454,7 +454,7 @@ class Transformer(nn.Module): 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([ torch.stack( torch.meshgrid(