shixian.shi
2023-08-16 7175dee6e19a4cbfc67ba83b6aefb93624eaffa7
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)