Skip to content

GitLab

Unverified Commit ae871ca9 authored by Andy Lo's avatar Andy Lo Committed by GitHub
Browse files

Minor cleanup for Voxtral (#34247) 


Signed-off-by: default avatarAndy Lo <andy@mistral.ai>
parent a2443de5
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
vllm/model_executor/models/voxtral.py
View file @ ae871ca9
...@@ -792,7 +792,9 @@ class VoxtralEncoderModel(nn.Module): ...@@ -792,7 +792,9 @@ class VoxtralEncoderModel(nn.Module):
audio_waveforms: torch.Tensor, audio_waveforms: torch.Tensor,
) -> torch.Tensor: ) -> torch.Tensor:
input_dtype = audio_waveforms.dtype 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( stft = torch.stft(
audio_waveforms, audio_waveforms,
self.config.window_size, self.config.window_size,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment