From 2fe3207d713f8d166e3cbaf1dc0a55aa7dd74784 Mon Sep 17 00:00:00 2001 From: chasonjiang <1440499136@qq.com> Date: Sat, 9 Mar 2024 22:11:07 +0800 Subject: [PATCH] modified: GPT_SoVITS/TTS_infer_pack/TTS.py --- GPT_SoVITS/TTS_infer_pack/TTS.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/GPT_SoVITS/TTS_infer_pack/TTS.py b/GPT_SoVITS/TTS_infer_pack/TTS.py index f54bf3b..7a8ecec 100644 --- a/GPT_SoVITS/TTS_infer_pack/TTS.py +++ b/GPT_SoVITS/TTS_infer_pack/TTS.py @@ -87,6 +87,7 @@ class TTS_Config: self.n_speakers:int = 300 self.langauges:list = ["auto", "en", "zh", "ja", "all_zh", "all_ja"] + print(self) def _load_configs(self, configs_path: str)->dict: with open(configs_path, 'r') as f: @@ -118,6 +119,18 @@ class TTS_Config: configs_path = self.configs_path with open(configs_path, 'w') as f: yaml.dump(configs, f) + + + def __str__(self): + string = "----------------TTS Config--------------\n" + string += "device: {}\n".format(self.device) + string += "is_half: {}\n".format(self.is_half) + string += "bert_base_path: {}\n".format(self.bert_base_path) + string += "t2s_weights_path: {}\n".format(self.t2s_weights_path) + string += "vits_weights_path: {}\n".format(self.vits_weights_path) + string += "cnhuhbert_base_path: {}\n".format(self.cnhuhbert_base_path) + string += "----------------------------------------\n" + return string class TTS: