[fix] [whisper] ensure inputs are moved to the correct device before processing. (#22293)

2026-04-08 23:45:42 +08:00
parent b5b2dbe05f
commit ae8da14ea3
1 changed files with 4 additions and 0 deletions
@@ -292,6 +292,10 @@ class WhisperEncoder(torch.nn.Module):
        position_ids: torch.Tensor,
        forward_batch: ForwardBatch,
    ):
+        device = self.conv1.weight.device
+        input_features = input_features.to(device=device)
+        position_ids = position_ids.to(device=device)
+
        inputs_embeds = torch.nn.functional.gelu(self.conv1(input_features))
        inputs_embeds = torch.nn.functional.gelu(self.conv2(inputs_embeds))