Fix _get_strided_batch device

lhotse-speech · Mar 11, 2024 · 50fd411 · 50fd411
1 parent 0673925
commit 50fd411
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lhotse/features/kaldi/layers.py b/lhotse/features/kaldi/layers.py
@@ -776,7 +776,7 @@ def _get_strided_batch(
         if npad_right >= 0:
             pad_right = torch.flip(waveform[:, -npad_right:], (1,))
         else:
-            pad_right = torch.zeros(0, dtype=waveform.dtype)
+            pad_right = torch.zeros(0, dtype=waveform.dtype, device=waveform.device)
         waveform = torch.cat((pad_left, waveform, pad_right), dim=1)
 
     strides = (