From 7c7d9ab04261a69a393768a0934cb6bf7f4f38af Mon Sep 17 00:00:00 2001 From: Zengwei Yao Date: Mon, 17 Apr 2023 12:03:52 +0800 Subject: [PATCH] add @torch.jit.export for streaming_forward func in Zipformer class (#1004) --- .../ASR/pruned_transducer_stateless7_streaming/zipformer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7_streaming/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7_streaming/zipformer.py index a5c422959..0a6886dec 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7_streaming/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7_streaming/zipformer.py @@ -570,6 +570,7 @@ class Zipformer(EncoderInterface): return x, lengths + @torch.jit.export def streaming_forward( self, x: torch.Tensor,