From ee57ff94dfa2c3ced30c1b103076b4ae18fa9199 Mon Sep 17 00:00:00 2001 From: espnetUser <81252087+espnetUser@users.noreply.github.com> Date: Tue, 10 May 2022 22:37:18 +0200 Subject: [PATCH] Update asr_inference_streaming.py Fix dtype CI error --- espnet2/bin/asr_inference_streaming.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/espnet2/bin/asr_inference_streaming.py b/espnet2/bin/asr_inference_streaming.py index ffd6e03c570..b2910a018ec 100755 --- 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