Skip to content

Commit bb7ea62

Browse files
committed
fix type issue
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
1 parent 5706beb commit bb7ea62

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/layers/rotary_embedding/llama4_vision_rope.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,6 @@ def forward_hip( # type: ignore[override]
8989
def forward_xpu( # type: ignore[override]
9090
self,
9191
query: torch.Tensor,
92-
key: Optional[torch.Tensor] = None,
93-
) -> tuple[torch.Tensor, Optional[torch.Tensor]]:
92+
key: torch.Tensor | None = None,
93+
) -> tuple[torch.Tensor, torch.Tensor | None]:
9494
return self.forward_native(query, key)

0 commit comments

Comments
 (0)