From 3b4fa4863f354bab10ea04a93c58700e0e2e2843 Mon Sep 17 00:00:00 2001
From: JinZr <60612200+JinZr@users.noreply.github.com>
Date: Tue, 25 Jul 2023 16:05:57 +0800
Subject: [PATCH] minor fixes

---
 egs/librispeech/ASR/zipformer_label_level_algn/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/zipformer_label_level_algn/model.py b/egs/librispeech/ASR/zipformer_label_level_algn/model.py
index d4ed9441c..c502224f6 100644
--- a/egs/librispeech/ASR/zipformer_label_level_algn/model.py
+++ b/egs/librispeech/ASR/zipformer_label_level_algn/model.py
@@ -265,7 +265,7 @@ class AsrModel(nn.Module):
         # project_input=False since we applied the decoder's input projections
         # prior to do_rnnt_pruning (this is an optimization for speed).
         logits = self.joiner(
-            am_pruned, lm_pruned, encoder_out_lens, project_input=False
+            am_pruned, lm_pruned, encoder_out_lens, apply_attn=True, project_input=False
         )
 
         with torch.cuda.amp.autocast(enabled=False):