Pārlūkot izejas kodu

Merge pull request #901 from alibaba-damo-academy/dev_lhn

fix onnx export bug
hnluo 2 gadi atpakaļ
vecāks
revīzija
b22bd68863
1 mainītis faili ar 3 papildinājumiem un 2 dzēšanām
  1. 3 2
      funasr/export/models/__init__.py

+ 3 - 2
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):