From 3724d5f6b59d9859e5b47c047535bb8edc124eab Mon Sep 17 00:00:00 2001 From: Chris <34248815+chrisociepa@users.noreply.github.com> Date: Sun, 15 Sep 2024 06:20:05 +0200 Subject: [PATCH] [Bugfix][Model] Fix Python 3.8 compatibility in Pixtral model by updating type annotations (#8490) --- vllm/model_executor/models/pixtral.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/models/pixtral.py b/vllm/model_executor/models/pixtral.py index b26fd558fa1e..682b78bbed09 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(