From 40b0b5365e319c4a0af420d5be4146e8be312f64 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Fri, 2 Jul 2021 10:45:00 +0200 Subject: [PATCH] Let `get_characters` return `num_chars` --- TTS/tts/models/base_tts.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/TTS/tts/models/base_tts.py b/TTS/tts/models/base_tts.py index 015d0200..2ec268d6 100644 --- a/TTS/tts/models/base_tts.py +++ b/TTS/tts/models/base_tts.py @@ -41,7 +41,8 @@ class BaseTTS(BaseModel): config.characters = parse_symbols() model_characters = phonemes if config.use_phonemes else symbols - return model_characters, config + num_chars = len(model_characters) + getattr(config, "add_blank", False) + return model_characters, config, num_chars def get_speaker_manager(config: Coqpit, restore_path: str, data: List, out_path: str = None) -> SpeakerManager: return get_speaker_manager(config, restore_path, data, out_path)