Update asr_inference_streaming.py

Fix CI error on mismatch in Tensor dtypes
chintu619 · May 11, 2022 · 87c7573 · 87c7573
1 parent 7a590cc
commit 87c7573
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
@@ -216,7 +216,7 @@ def apply_frontend(
         if not has_enough_samples:
             if is_final:
                 pad = torch.zeros(
-                    self.win_length - speech.size(0), dtype=getattr(torch, self.dtype)
+                    self.win_length - speech.size(0), dtype=speech.dtype
                 )
                 speech = torch.cat([speech, pad], dim=0)
             else: