| funasr/export/export_model.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/export/export_model.py
@@ -254,7 +254,7 @@ 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,