From 72b6c7735e44fff12c4a3d401deb98c59f16ab0f Mon Sep 17 00:00:00 2001 From: hnluo Date: Tue, 29 Aug 2023 11:40:46 +0800 Subject: [PATCH] fix onnx export bug --- funasr/export/models/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/funasr/export/models/__init__.py b/funasr/export/models/__init__.py index cba92a865..94447dca3 100644 --- a/funasr/export/models/__init__.py +++ b/funasr/export/models/__init__.py @@ -25,8 +25,9 @@ def get_model(model, export_config=None): 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):