From ec12fce1ba07b18736448d2aa99a758b8265b804 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期三, 15 五月 2024 14:50:59 +0800
Subject: [PATCH] update run_server.sh
---
runtime/run_server.sh | 5 +++++
1 files changed, 5 insertions(+), 0 deletions(-)
diff --git a/runtime/run_server.sh b/runtime/run_server.sh
index 22907ab..bee4f61 100644
--- a/runtime/run_server.sh
+++ b/runtime/run_server.sh
@@ -1,4 +1,8 @@
+TORCH_DIR=$(python3 -c "import torch; import os; print(os.path.dirname(torch.__file__))")
+BLADE_DIR=$(python3 -c "import torch_blade; import os; print(os.path.dirname(torch_blade.__file__))")
+export LD_LIBRARY_PATH=/usr/local/lib:${TORCH_DIR}/lib:${BLADE_DIR}:${LD_LIBRARY_PATH}
+
download_model_dir="/workspace/models"
model_dir="damo/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-onnx"
vad_dir="damo/speech_fsmn_vad_zh-cn-16k-common-onnx"
@@ -38,5 +42,6 @@
--port ${port} \
--certfile "${certfile}" \
--keyfile "${keyfile}" \
+ --gpu \
--hotword "${hotword}" &
--
Gitblit v1.9.1