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

Fixup

This commit is contained in:
Eren G??lge 2023-05-24 12:13:40 +02:00
parent 659bce0d6c
commit af16348cfb
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,6 +63,18 @@ class BaseVocabulary:
the vocabulary.""" the vocabulary."""
return self.char_to_id(self.blank) if self.blank else len(self.vocab) return self.char_to_id(self.blank) if self.blank else len(self.vocab)
@property
def bos_id(self) -> int:
"""Return the index of the bos character. If the bos character is not specified, return the length of the
vocabulary."""
return self.char_to_id(self.bos) if self.bos else len(self.vocab)
@property
def eos_id(self) -> int:
"""Return the index of the eos character. If the eos character is not specified, return the length of the
vocabulary."""
return self.char_to_id(self.eos) if self.eos else len(self.vocab)
@property @property
def vocab(self): def vocab(self):
"""Return the vocabulary dictionary.""" """Return the vocabulary dictionary."""
@ -187,6 +199,14 @@ class BaseCharacters:
def blank_id(self) -> int: def blank_id(self) -> int:
return self.char_to_id(self.blank) if self.blank else len(self.vocab) return self.char_to_id(self.blank) if self.blank else len(self.vocab)
@property
def eos_id(self) -> int:
return self.char_to_id(self.eos) if self.eos else len(self.vocab)
@property
def bos_id(self) -> int:
return self.char_to_id(self.bos) if self.bos else len(self.vocab)
@property @property
def characters(self): def characters(self):
return self._characters return self._characters