From c4882b43fce3c32cb0ce3c9fc2c164f0ce0e8213 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 30 三月 2023 16:04:13 +0800
Subject: [PATCH] export
---
funasr/export/export_model.py | 7 +++++++
1 files changed, 7 insertions(+), 0 deletions(-)
diff --git a/funasr/export/export_model.py b/funasr/export/export_model.py
index 9afa7b1..444ccf4 100644
--- a/funasr/export/export_model.py
+++ b/funasr/export/export_model.py
@@ -205,6 +205,13 @@
model, punc_train_args = PUNCTask.build_model_from_file(
punc_train_config, punc_model_file, 'cpu'
)
+ elif mode.startswith('punc_VadRealtime'):
+ from funasr.tasks.punctuation import PunctuationTask as PUNCTask
+ punc_train_config = os.path.join(model_dir, 'config.yaml')
+ punc_model_file = os.path.join(model_dir, 'punc.pb')
+ model, punc_train_args = PUNCTask.build_model_from_file(
+ punc_train_config, punc_model_file, 'cpu'
+ )
self._export(model, tag_name)
--
Gitblit v1.9.1