Update asr_inference_streaming.py

Fix dtype CI error
chintu619 · May 10, 2022 · ee57ff9 · ee57ff9
1 parent 272d5d0
commit ee57ff9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espnet2/bin/asr_inference_streaming.py b/espnet2/bin/asr_inference_streaming.py
@@ -215,7 +215,7 @@ def apply_frontend(
         has_enough_samples = False if speech.size(0) <= self.win_length else True
         if not has_enough_samples:
             if is_final:
-                pad = torch.zeros(self.win_length - speech.size(0))
+                pad = torch.zeros(self.win_length - speech.size(0), dtype=getattr(torch, self.dtype))
                 speech = torch.cat([speech, pad], dim=0)
             else:
                 feats = None