diff --git a/funasr/export/export_model.py b/funasr/export/export_model.py index e0a9313ed..6ab94084c 100644 --- a/funasr/export/export_model.py +++ b/funasr/export/export_model.py @@ -254,7 +254,7 @@ class ModelExport: if not os.path.exists(quant_model_path): onnx_model = onnx.load(model_path) nodes = [n.name for n in onnx_model.graph.node] - nodes_to_exclude = [m for m in nodes if 'output' in m] + nodes_to_exclude = [m for m in nodes if 'output' in m or 'bias_encoder' in m or 'bias_decoder' in m] quantize_dynamic( model_input=model_path, model_output=quant_model_path,