From 31eed1834f9ff17d6246008f64d3e061f58ef80a Mon Sep 17 00:00:00 2001
From: 凌匀 <ailsa.zly@alibaba-inc.com>
Date: 星期一, 27 二月 2023 13:33:55 +0800
Subject: [PATCH] in_cache & support soundfile read
---
funasr/export/export_model.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/export/export_model.py b/funasr/export/export_model.py
index e1d5fdb..933a927 100644
--- a/funasr/export/export_model.py
+++ b/funasr/export/export_model.py
@@ -58,7 +58,7 @@
if enc_size:
dummy_input = model.get_dummy_inputs(enc_size)
else:
- dummy_input = model.get_dummy_inputs_txt()
+ dummy_input = model.get_dummy_inputs()
# model_script = torch.jit.script(model)
model_script = torch.jit.trace(model, dummy_input)
@@ -111,7 +111,7 @@
dummy_input,
os.path.join(path, f'{model.model_name}.onnx'),
verbose=verbose,
- opset_version=12,
+ opset_version=14,
input_names=model.get_input_names(),
output_names=model.get_output_names(),
dynamic_axes=model.get_dynamic_axes()
--
Gitblit v1.9.1