From c1df24ea95eb09ca533fba9914bc8349d5dd8113 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 08 七月 2024 11:38:03 +0800
Subject: [PATCH] wechat
---
runtime/run_server_2pass.sh | 51 +++++++++++++++++++++++----------------------------
1 files changed, 23 insertions(+), 28 deletions(-)
diff --git a/runtime/run_server_2pass.sh b/runtime/run_server_2pass.sh
index e017708..bb33171 100644
--- a/runtime/run_server_2pass.sh
+++ b/runtime/run_server_2pass.sh
@@ -1,49 +1,44 @@
download_model_dir="/workspace/models"
-model_dir="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-onnx"
+model_dir="damo/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-onnx"
online_model_dir="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-online-onnx"
vad_dir="damo/speech_fsmn_vad_zh-cn-16k-common-onnx"
punc_dir="damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727-onnx"
itn_dir="thuduj12/fst_itn_zh"
-decoder_thread_num=32
-model_thread_num=1
-io_thread_num=8
+lm_dir="damo/speech_ngram_lm_zh-cn-ai-wesp-fst"
port=10095
-certfile="../../../ssl_key/server.crt"
-keyfile="../../../ssl_key/server.key"
-hotword="../../hotwords.txt"
+certfile="$(pwd)/ssl_key/server.crt"
+keyfile="$(pwd)/ssl_key/server.key"
+hotword="$(pwd)/websocket/hotwords.txt"
+# set decoder_thread_num
+decoder_thread_num=$(cat /proc/cpuinfo | grep "processor"|wc -l) || { echo "Get cpuinfo failed. Set decoder_thread_num = 32"; decoder_thread_num=32; }
+multiple_io=16
+io_thread_num=$(( (decoder_thread_num + multiple_io - 1) / multiple_io ))
+model_thread_num=1
+cmd_path=/workspace/FunASR/runtime/websocket/build/bin
+cmd=funasr-wss-server-2pass
-. ../egs/aishell/transformer/utils/parse_options.sh || exit 1;
+. ./tools/utils/parse_options.sh || exit 1;
-cd /workspace/FunASR/runtime/websocket/build/bin
-if [ -z "$certfile" ] || [ "$certfile" -eq 0 ]; then
-./funasr-wss-server-2pass \
+if [ -z "$certfile" ] || [ "$certfile" = "0" ]; then
+ certfile=""
+ keyfile=""
+fi
+
+cd $cmd_path
+$cmd_path/${cmd} \
--download-model-dir "${download_model_dir}" \
--model-dir "${model_dir}" \
--online-model-dir "${online_model_dir}" \
--vad-dir "${vad_dir}" \
--punc-dir "${punc_dir}" \
--itn-dir "${itn_dir}" \
- --decoder-thread-num ${decoder_thread_num} \
- --model-thread-num ${model_thread_num} \
- --io-thread-num ${io_thread_num} \
- --port ${port} \
- --certfile "" \
- --keyfile "" \
- --hotword "${hotword}"
-else
-./funasr-wss-server-2pass \
- --download-model-dir "${download_model_dir}" \
- --model-dir "${model_dir}" \
- --online-model-dir "${online_model_dir}" \
- --vad-dir "${vad_dir}" \
- --punc-dir "${punc_dir}" \
- --itn-dir "${itn_dir}" \
+ --lm-dir "${lm_dir}" \
--decoder-thread-num ${decoder_thread_num} \
--model-thread-num ${model_thread_num} \
--io-thread-num ${io_thread_num} \
--port ${port} \
--certfile "${certfile}" \
--keyfile "${keyfile}" \
- --hotword "${hotword}"
-fi
+ --hotword "${hotword}" &
+
--
Gitblit v1.9.1