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

View File

@ -283,12 +283,9 @@ class IPAPhonemes(BaseCharacters):
@staticmethod
def init_from_config(config: "Coqpit"):
# band-aid for compatibility with old models
characters = None
if "characters" in config:
if "phonemes" in config.characters:
if "characters" in config and config.characters is not None:
if "phonemes" in config.characters and config.characters.phonemes is not None:
config.characters["characters"] = config.characters["phonemes"]
# delattr(config.characters, "phonemes")
return IPAPhonemes(
characters=config.characters["characters"],
punctuations=config.characters["punctuations"],
@ -299,7 +296,10 @@ class IPAPhonemes(BaseCharacters):
is_unique=config.characters["is_unique"],
is_sorted=config.characters["is_sorted"],
)
return characters
else:
return IPAPhonemes(
**config.characters if config.characters is not None else {},
)
class Graphemes(BaseCharacters):