Update decode.py

2025-09-05 23:24:17 +00:00 · 2023-11-09 11:37:18 +08:00 · 2023-11-09 11:37:18 +08:00 · 16499a5ef6
commit 16499a5ef6
parent fb541ec60c
1 changed files with 1 additions and 0 deletions
--- a/egs/multi_zh-hans/ASR/zipformer/decode.py
+++ b/egs/multi_zh-hans/ASR/zipformer/decode.py
@ -520,6 +520,7 @@ def decode_one_batch(
            encoder_out=encoder_out,
            encoder_out_lens=encoder_out_lens,
            beam=params.beam_size,
+            context_graph=context_graph,
        )
        for hyp in sp.decode(hyp_tokens):
            hyps.append(hyp.split())