Skip to content

Commit

Permalink
resolve PR issue
Browse files Browse the repository at this point in the history
  • Loading branch information
root authored and root committed Aug 14, 2024
1 parent 8d0107a commit 5632925
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion egs/reazonspeech/ASR/zipformer/streaming_decode.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 5632925

Please sign in to comment.