Fix: disable multiprocessing for bert_gen due to pyopenjtalk_worker

2024-03-13 17:12:12 +09:00
parent b71e304fe9
commit 67f19ba627
3 changed files with 11 additions and 9 deletions
--- a/config.py
+++ b/config.py
@@ -92,7 +92,7 @@ class Bert_gen_config:
    def __init__(
        self,
        config_path: str,
-        num_processes: int = 2,
+        num_processes: int = 1,
        device: str = "cuda",
        use_multi_device: bool = False,
    ):