shixian.shi
2023-09-12 b883a7825680352d0927f4dcb9f3a7807ed0ac63
update contextual export
1个文件已修改
2 ■■■ 已修改文件
funasr/export/export_model.py 2 ●●● 补丁 | 查看 | 原始文档 | 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,