From d781aa27e16e25ec04b66bab17b735e56005b3b7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 23 三月 2023 15:47:04 +0800
Subject: [PATCH] websocket

---
 funasr/runtime/python/websocket/ASR_server.py |    4 ++++
 1 files changed, 4 insertions(+), 0 deletions(-)

diff --git a/funasr/runtime/python/websocket/ASR_server.py b/funasr/runtime/python/websocket/ASR_server.py
index 0aed2b4..ac63d3c 100644
--- a/funasr/runtime/python/websocket/ASR_server.py
+++ b/funasr/runtime/python/websocket/ASR_server.py
@@ -38,6 +38,10 @@
                     type=str,
                     default="",
                     help="model from modelscope")
+parser.add_argument("--ngpu",
+                    type=int,
+                    default=1,
+                    help="0 for cpu, 1 for gpu")
 
 args = parser.parse_args()
 

--
Gitblit v1.9.1