speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update contextual export

Browse Source
This commit is contained in:
shixian.shi 2023-09-12 19:40:02 +08:00
parent 13c2af5de4
commit b883a78256
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/export/export_model.py
View File

@ -254,7 +254,7 @@ class ModelExport:
if not os.path.exists(quant_model_path): if not os.path.exists(quant_model_path):
onnx_model = onnx.load(model_path) onnx_model = onnx.load(model_path)
nodes = [n.name for n in onnx_model.graph.node] 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( quantize_dynamic(
model_input=model_path, model_input=model_path,
model_output=quant_model_path, model_output=quant_model_path,