From 8fbd79e3c03fb8cef214d0c3060f3b21c4dcd4e5 Mon Sep 17 00:00:00 2001
From: jinzr <60612200+JinZr@users.noreply.github.com>
Date: Mon, 23 Oct 2023 21:38:33 +0800
Subject: [PATCH] Revert "Update decoder.py"

This reverts commit 0c8b3975170f74b61e98885771eccd7f94679608.
---
 egs/libriheavy/ASR/zipformer_prompt_asr/decoder.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/egs/libriheavy/ASR/zipformer_prompt_asr/decoder.py b/egs/libriheavy/ASR/zipformer_prompt_asr/decoder.py
index 8a55eb5c8..93e0f9f7e 100644
--- a/egs/libriheavy/ASR/zipformer_prompt_asr/decoder.py
+++ b/egs/libriheavy/ASR/zipformer_prompt_asr/decoder.py
@@ -95,10 +95,6 @@ class Decoder(nn.Module):
                 max_abs=1.0,
                 prob=0.05,
             )
-        else:
-            # To avoid `RuntimeError: Module 'Decoder' has no attribute 'conv'`
-            # when inference with torch.jit.script and context_size == 1
-            self.conv = nn.Identity()
 
     def forward(self, y: torch.Tensor, need_pad: bool = True) -> torch.Tensor:
         """