From 563292599b286f00cfcaebe8022e9a5f1b829dcf Mon Sep 17 00:00:00 2001 From: root Date: Thu, 1 Aug 2024 18:27:34 +0900 Subject: [PATCH] resolve PR issue --- egs/reazonspeech/ASR/zipformer/streaming_decode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/reazonspeech/ASR/zipformer/streaming_decode.py b/egs/reazonspeech/ASR/zipformer/streaming_decode.py index 693c8db1c5..81bdc4845b 100755 --- a/egs/reazonspeech/ASR/zipformer/streaming_decode.py +++ b/egs/reazonspeech/ASR/zipformer/streaming_decode.py @@ -490,7 +490,7 @@ def decode_one_chunk( if params.decoding_method == "greedy_search": greedy_search(model=model, encoder_out=encoder_out, streams=decode_streams) elif params.decoding_method == "fast_beam_search": - processed_lens = torch.tensor(processed_lens, device=device) + processed_lens = torch.tensor(processed_lens, device=model.device) processed_lens = processed_lens + encoder_out_lens fast_beam_search_one_best( model=model,