tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix: pyopenjtalk_worker not working

Browse Source
This commit is contained in:
tsukumi
2024-03-08 09:21:09 +00:00
parent df68771651
commit 766699e812
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
style_bert_vits2/nlp/japanese/pyopenjtalk_worker/__init__.py
View File

@@ -56,7 +56,6 @@ def initialize(port: int = WORKER_PORT) -> None:
import sys import sys
import time import time
logger.debug("initialize")
global WORKER_CLIENT global WORKER_CLIENT
if WORKER_CLIENT: if WORKER_CLIENT:
return return
@@ -98,6 +97,7 @@ def initialize(port: int = WORKER_PORT) -> None:
if count == 20: if count == 20:
raise TimeoutError("サーバーに接続できませんでした") raise TimeoutError("サーバーに接続できませんでした")
logger.debug("pyopenjtalk worker server started")
WORKER_CLIENT = client WORKER_CLIENT = client
atexit.register(terminate) atexit.register(terminate)
@@ -110,7 +110,7 @@ def initialize(port: int = WORKER_PORT) -> None:
# top-level declaration # top-level declaration
def terminate() -> None: def terminate() -> None:
logger.debug("terminate") logger.debug("pyopenjtalk worker server terminated")
global WORKER_CLIENT global WORKER_CLIENT
if not WORKER_CLIENT: if not WORKER_CLIENT:
return return

4
style_bert_vits2/nlp/japanese/pyopenjtalk_worker/worker_server.py
View File

@@ -16,7 +16,7 @@ from style_bert_vits2.nlp.japanese.pyopenjtalk_worker.worker_common import (
# To make it as fast as possible # To make it as fast as possible
# Probably faster than calling getattr every time # Probably faster than calling getattr every time
__PYOPENJTALK_FUNC_DICT = { PYOPENJTALK_FUNC_DICT = {
"run_frontend": pyopenjtalk.run_frontend, "run_frontend": pyopenjtalk.run_frontend,
"make_label": pyopenjtalk.make_label, "make_label": pyopenjtalk.make_label,
"mecab_dict_index": pyopenjtalk.mecab_dict_index, "mecab_dict_index": pyopenjtalk.mecab_dict_index,
@@ -57,7 +57,7 @@ class WorkerServer:
elif request_type == RequestType.PYOPENJTALK: elif request_type == RequestType.PYOPENJTALK:
func_name = request.get("func") func_name = request.get("func")
assert isinstance(func_name, str) assert isinstance(func_name, str)
func = __PYOPENJTALK_FUNC_DICT[func_name] func = PYOPENJTALK_FUNC_DICT[func_name]
args = request.get("args") args = request.get("args")
kwargs = request.get("kwargs") kwargs = request.get("kwargs")
assert isinstance(args, list) assert isinstance(args, list)