From 9dff68e34d0fdc540b8fc0932be070dd21a68182 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=82=A6?= Date: Sun, 27 Oct 2024 13:29:36 +0800 Subject: [PATCH] Update api_v2.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 支持接口多进程 --- api_v2.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api_v2.py b/api_v2.py index 92a18f37..6a9c6607 100644 --- a/api_v2.py +++ b/api_v2.py @@ -128,11 +128,13 @@ parser = argparse.ArgumentParser(description="GPT-SoVITS api") parser.add_argument("-c", "--tts_config", type=str, default="GPT_SoVITS/configs/tts_infer.yaml", help="tts_infer路径") parser.add_argument("-a", "--bind_addr", type=str, default="127.0.0.1", help="default: 127.0.0.1") parser.add_argument("-p", "--port", type=int, default="9880", help="default: 9880") +parser.add_argument("-w", "--workers", type=int, default="1", help="default: 1") args = parser.parse_args() config_path = args.tts_config # device = args.device port = args.port host = args.bind_addr +workers = args.workers argv = sys.argv if config_path in [None, ""]: @@ -453,7 +455,7 @@ if __name__ == "__main__": try: if host == 'None': # 在调用时使用 -a None 参数，可以让api监听双栈 host = None - uvicorn.run(app=APP, host=host, port=port, workers=1) + uvicorn.run(app="api_v2:APP", host=host, port=port, workers=workers) except Exception as e: traceback.print_exc() os.kill(os.getpid(), signal.SIGTERM)