From b5d3df75cf6462aa3bf42fd3c86fa2aa7f1c8a15 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 24 十一月 2023 00:54:44 +0800
Subject: [PATCH] setup jamo
---
funasr/export/models/__init__.py | 11 ++++++-----
1 files changed, 6 insertions(+), 5 deletions(-)
diff --git a/funasr/export/models/__init__.py b/funasr/export/models/__init__.py
index cba92a8..b7b0889 100644
--- a/funasr/export/models/__init__.py
+++ b/funasr/export/models/__init__.py
@@ -1,7 +1,7 @@
from funasr.models.e2e_asr_paraformer import Paraformer, BiCifParaformer, ParaformerOnline
from funasr.export.models.e2e_asr_paraformer import Paraformer as Paraformer_export
from funasr.export.models.e2e_asr_paraformer import BiCifParaformer as BiCifParaformer_export
-from funasr.export.models.e2e_asr_conformer import Conformer as Conformer_export
+# from funasr.export.models.e2e_asr_conformer import Conformer as Conformer_export
from funasr.models.e2e_vad import E2EVadModel
from funasr.export.models.e2e_vad import E2EVadModel as E2EVadModel_export
@@ -25,12 +25,13 @@
elif isinstance(model, BiCifParaformer):
return BiCifParaformer_export(model, **export_config)
elif isinstance(model, ParaformerOnline):
- return (ParaformerOnline_encoder_predictor_export(model, model_name="model"),
- ParaformerOnline_decoder_export(model, model_name="decoder"))
+ encoder = ParaformerOnline_encoder_predictor_export(model, model_name="model")
+ decoder = ParaformerOnline_decoder_export(model, model_name="decoder")
+ return [encoder, decoder]
elif isinstance(model, Paraformer):
return Paraformer_export(model, **export_config)
- elif isinstance(model, Conformer_export):
- return Conformer_export(model, **export_config)
+ # elif isinstance(model, Conformer_export):
+ # return Conformer_export(model, **export_config)
elif isinstance(model, E2EVadModel):
return E2EVadModel_export(model, **export_config)
elif isinstance(model, PunctuationModel):
--
Gitblit v1.9.1