From 762ca9e38a3a06bbc1bdb0c909ba5d0a521754f8 Mon Sep 17 00:00:00 2001 From: NickLucche Date: Wed, 24 Dec 2025 14:29:36 +0000 Subject: [PATCH] signature Signed-off-by: NickLucche --- vllm/model_executor/models/whisper.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/vllm/model_executor/models/whisper.py b/vllm/model_executor/models/whisper.py index abc903292d9f2..06376c71f33ec 100644 --- a/vllm/model_executor/models/whisper.py +++ b/vllm/model_executor/models/whisper.py @@ -534,9 +534,7 @@ class WhisperEncoder(nn.Module): sinusoids(*self.embed_positions.weight.shape) ) - def forward_conv( - self, input_features: torch.Tensor | list[torch.Tensor] - ) -> torch.Tensor: + def forward_conv(self, input_features: torch.Tensor) -> torch.Tensor: embeds = nn.functional.gelu(self.conv1(input_features)) embeds = nn.functional.gelu(self.conv2(embeds)) embeds = embeds.transpose(-1, -2) @@ -557,7 +555,7 @@ class WhisperEncoder(nn.Module): hidden_states = self.layer_norm(hidden_states) return hidden_states - def forward(self, input_features: torch.Tensor | list[torch.Tensor]): + def forward(self, input_features: torch.Tensor): hidden_states = self.forward_conv(input_features) return self.forward_layers(hidden_states)