Minor cleanup for Voxtral (#34247)

Signed-off-by: Andy Lo <andy@mistral.ai>
2026-02-10 18:18:30 +00:00
parent a2443de5fa
commit ae871ca923
1 changed files with 3 additions and 1 deletions
--- a/vllm/model_executor/models/voxtral.py
+++ b/vllm/model_executor/models/voxtral.py
@@ -792,7 +792,9 @@ class VoxtralEncoderModel(nn.Module):
        audio_waveforms: torch.Tensor,
    ) -> torch.Tensor:
        input_dtype = audio_waveforms.dtype
-        window = torch.hann_window(self.config.window_size).to(audio_waveforms.device)
+        window = torch.hann_window(
+            self.config.window_size, device=audio_waveforms.device
+        )
        stft = torch.stft(
            audio_waveforms,
            self.config.window_size,