From 766699e812084322822027dee76b1003dbe02918 Mon Sep 17 00:00:00 2001 From: tsukumi Date: Fri, 8 Mar 2024 09:21:09 +0000 Subject: [PATCH] Fix: pyopenjtalk_worker not working --- style_bert_vits2/nlp/japanese/pyopenjtalk_worker/__init__.py | 4 ++-- .../nlp/japanese/pyopenjtalk_worker/worker_server.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/__init__.py b/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/__init__.py index 90419f5..670461d 100644 --- a/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/__init__.py +++ b/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/__init__.py @@ -56,7 +56,6 @@ def initialize(port: int = WORKER_PORT) -> None: import sys import time - logger.debug("initialize") global WORKER_CLIENT if WORKER_CLIENT: return @@ -98,6 +97,7 @@ def initialize(port: int = WORKER_PORT) -> None: if count == 20: raise TimeoutError("サーバーに接続できませんでした") + logger.debug("pyopenjtalk worker server started") WORKER_CLIENT = client atexit.register(terminate) @@ -110,7 +110,7 @@ def initialize(port: int = WORKER_PORT) -> None: # top-level declaration def terminate() -> None: - logger.debug("terminate") + logger.debug("pyopenjtalk worker server terminated") global WORKER_CLIENT if not WORKER_CLIENT: return diff --git a/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/worker_server.py b/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/worker_server.py index e9a3dc1..149323a 100644 --- a/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/worker_server.py +++ b/style_bert_vits2/nlp/japanese/pyopenjtalk_worker/worker_server.py @@ -16,7 +16,7 @@ from style_bert_vits2.nlp.japanese.pyopenjtalk_worker.worker_common import ( # To make it as fast as possible # Probably faster than calling getattr every time -__PYOPENJTALK_FUNC_DICT = { +PYOPENJTALK_FUNC_DICT = { "run_frontend": pyopenjtalk.run_frontend, "make_label": pyopenjtalk.make_label, "mecab_dict_index": pyopenjtalk.mecab_dict_index, @@ -57,7 +57,7 @@ class WorkerServer: elif request_type == RequestType.PYOPENJTALK: func_name = request.get("func") assert isinstance(func_name, str) - func = __PYOPENJTALK_FUNC_DICT[func_name] + func = PYOPENJTALK_FUNC_DICT[func_name] args = request.get("args") kwargs = request.get("kwargs") assert isinstance(args, list)