From 6850e9f86a2c478455907eb575748eb9c45cbddc Mon Sep 17 00:00:00 2001
From: nichongjia-2007 <nichongjia@gmail.com>
Date: 星期二, 11 七月 2023 17:36:11 +0800
Subject: [PATCH] add export conformer
---
funasr/export/models/e2e_asr_conformer.py | 3 +--
1 files changed, 1 insertions(+), 2 deletions(-)
diff --git a/funasr/export/models/e2e_asr_conformer.py b/funasr/export/models/e2e_asr_conformer.py
index 49c9aae..69907fb 100644
--- a/funasr/export/models/e2e_asr_conformer.py
+++ b/funasr/export/models/e2e_asr_conformer.py
@@ -6,8 +6,7 @@
from funasr.export.utils.torch_function import sequence_mask
from funasr.models.encoder.conformer_encoder import ConformerEncoder
from funasr.export.models.encoder.conformer_encoder import ConformerEncoder as ConformerEncoder_export
-from funasr.models.decoder.transformer_decoder import TransformerDecoder as TransformerDecoder_export
-
+from funasr.export.models.decoder.xformer_decoder import XformerDecoder as TransformerDecoder_export
class Conformer(nn.Module):
"""
--
Gitblit v1.9.1