Let `get_characters` return `num_chars`

This commit is contained in:
Eren Gölge 2021-07-02 10:45:00 +02:00
parent 0fa6a8c9b8
commit 40b0b5365e
1 changed files with 2 additions and 1 deletions

View File

@ -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)