shixian.shi
2023-08-16 6f753a3899891f129a0d8f7ea11646f3ea6dfe31
funasr/export/models/e2e_asr_contextual_paraformer.py
@@ -55,7 +55,7 @@
            self.decoder = ParaformerDecoderSAN_export(model.decoder, onnx=onnx)
        
        self.feats_dim = feats_dim
        self.model_name = model_name + '_bb'
        self.model_name = model_name
        if onnx:
            self.make_pad_mask = MakePadMask(max_seq_len, flip=False)