Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix IPAPhonemes init_from_config

This commit is contained in:
Eren Gölge 2021-11-25 17:30:54 +01:00
parent 0fe39166fe
commit 4e8f9d6f10
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
TTS/tts/utils/text/characters.py
View File

@ -209,7 +209,7 @@ class BaseCharacters:
), f" [!] There are duplicate characters in the character set. {set([x for x in self.vocab if self.vocab.count(x) > 1])}" ), f" [!] There are duplicate characters in the character set. {set([x for x in self.vocab if self.vocab.count(x) > 1])}"
def char_to_id(self, char: str) -> int: def char_to_id(self, char: str) -> int:
return self._char_to_id[char] return self._char_to_id[char]
def id_to_char(self, idx: int) -> str: def id_to_char(self, idx: int) -> str:
return self._id_to_char[idx] return self._id_to_char[idx]
@ -283,12 +283,9 @@ class IPAPhonemes(BaseCharacters):
@staticmethod @staticmethod
def init_from_config(config: "Coqpit"): def init_from_config(config: "Coqpit"):
# band-aid for compatibility with old models # band-aid for compatibility with old models
characters = None if "characters" in config and config.characters is not None:
if "characters" in config: if "phonemes" in config.characters and config.characters.phonemes is not None:
if "phonemes" in config.characters:
config.characters["characters"] = config.characters["phonemes"] config.characters["characters"] = config.characters["phonemes"]
# delattr(config.characters, "phonemes")
return IPAPhonemes( return IPAPhonemes(
characters=config.characters["characters"], characters=config.characters["characters"],
punctuations=config.characters["punctuations"], punctuations=config.characters["punctuations"],
@ -299,7 +296,10 @@ class IPAPhonemes(BaseCharacters):
is_unique=config.characters["is_unique"], is_unique=config.characters["is_unique"],
is_sorted=config.characters["is_sorted"], is_sorted=config.characters["is_sorted"],
) )
return characters else:
return IPAPhonemes(
**config.characters if config.characters is not None else {},
)
class Graphemes(BaseCharacters): class Graphemes(BaseCharacters):