Skip to content

Tokenizer ipa

TokenizerIPA

DEPRECATED: TokenizerIPA is a class for tokenizing International Phonetic Alphabet (IPA) phonemes.

Attributes:

Name Type Description
lang str

Language to be used. Default is "en".

phonemizer_checkpoint str

Path to the phonemizer checkpoint file.

phonemizer Phonemizer

Phonemizer object for converting text to phonemes.

tokenizer SequenceTokenizer

SequenceTokenizer object for tokenizing the phonemes.

Source code in training/preprocess/tokenizer_ipa.py
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
class TokenizerIPA:
    r"""DEPRECATED: TokenizerIPA is a class for tokenizing International Phonetic Alphabet (IPA) phonemes.

    Attributes:
        lang (str): Language to be used. Default is "en".
        phonemizer_checkpoint (str): Path to the phonemizer checkpoint file.
        phonemizer (Phonemizer): Phonemizer object for converting text to phonemes.
        tokenizer (SequenceTokenizer): SequenceTokenizer object for tokenizing the phonemes.
    """

    def __init__(
        self,
        lang: str = "en",
        phonemizer_checkpoint: str = "checkpoints/en_us_cmudict_ipa_forward.pt",
    ):
        r"""Initializes TokenizerIPA with the given language and phonemizer checkpoint.

        Args:
            lang (str): The language to be used. Default is "en".
            phonemizer_checkpoint (str): The path to the phonemizer checkpoint file.
        """
        lang_map = get_lang_map(lang)
        self.lang = lang_map.phonemizer

        self.phonemizer = Phonemizer.from_checkpoint(phonemizer_checkpoint)

        phoneme_symbols = [
            # IPA symbols
            "a",
            "b",
            "d",
            "e",
            "f",
            "g",
            "h",
            "i",
            "j",
            "k",
            "l",
            "m",
            "n",
            "o",
            "p",
            "r",
            "s",
            "t",
            "u",
            "v",
            "w",
            "x",
            "y",
            "z",
            "æ",
            "ç",
            "ð",
            "ø",
            "ŋ",
            "œ",
            "ɐ",
            "ɑ",
            "ɔ",
            "ə",
            "ɛ",
            "ɝ",
            "ɹ",
            "ɡ",
            "ɪ",
            "ʁ",
            "ʃ",
            "ʊ",
            "ʌ",
            "ʏ",
            "ʒ",
            "ʔ",
            "ˈ",
            "ˌ",
            "ː",
            "̃",
            "̍",
            "̥",
            "̩",
            "̯",
            "͡",
            "θ",
            # Punctuation
            "!",
            "?",
            ",",
            ".",
            "-",
            ":",
            ";",
            '"',
            "'",
            "(",
            ")",
            " ",
        ]

        self.tokenizer = SequenceTokenizer(
            phoneme_symbols,
            languages=["de", "en_us"],
            lowercase=True,
            char_repeats=1,
            append_start_end=True,
        )

        # test_text = "Hello, World!"
        # print("Initializing TokenizerIPA, check on: ", test_text)

        # phones_ipa = self.phonemizer(test_text, lang=self.lang)
        # tokens = self.tokenizer(phones_ipa, language=self.lang)

        # print("phones_ipa: ", phones_ipa)
        # print("tokens: ", tokens)
        # print("decoded: ", self.tokenizer.decode(tokens))

    def __call__(self, text: str) -> Tuple[Union[str, List[str]], List[int]]:
        r"""Converts the input text to phonemes and tokenizes them.

        Args:
            text (str): The input text to be tokenized.

        Returns:
            list: The tokenized phonemes.

        """
        phones_ipa = self.phonemizer(text, lang=self.lang)
        tokens = self.tokenizer(phones_ipa, language=self.lang)

        return phones_ipa, tokens

__call__(text)

Converts the input text to phonemes and tokenizes them.

Parameters:

Name Type Description Default
text str

The input text to be tokenized.

required

Returns:

Name Type Description
list Tuple[Union[str, List[str]], List[int]]

The tokenized phonemes.

Source code in training/preprocess/tokenizer_ipa.py
126
127
128
129
130
131
132
133
134
135
136
137
138
139
def __call__(self, text: str) -> Tuple[Union[str, List[str]], List[int]]:
    r"""Converts the input text to phonemes and tokenizes them.

    Args:
        text (str): The input text to be tokenized.

    Returns:
        list: The tokenized phonemes.

    """
    phones_ipa = self.phonemizer(text, lang=self.lang)
    tokens = self.tokenizer(phones_ipa, language=self.lang)

    return phones_ipa, tokens

__init__(lang='en', phonemizer_checkpoint='checkpoints/en_us_cmudict_ipa_forward.pt')

Initializes TokenizerIPA with the given language and phonemizer checkpoint.

Parameters:

Name Type Description Default
lang str

The language to be used. Default is "en".

'en'
phonemizer_checkpoint str

The path to the phonemizer checkpoint file.

'checkpoints/en_us_cmudict_ipa_forward.pt'
Source code in training/preprocess/tokenizer_ipa.py
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
def __init__(
    self,
    lang: str = "en",
    phonemizer_checkpoint: str = "checkpoints/en_us_cmudict_ipa_forward.pt",
):
    r"""Initializes TokenizerIPA with the given language and phonemizer checkpoint.

    Args:
        lang (str): The language to be used. Default is "en".
        phonemizer_checkpoint (str): The path to the phonemizer checkpoint file.
    """
    lang_map = get_lang_map(lang)
    self.lang = lang_map.phonemizer

    self.phonemizer = Phonemizer.from_checkpoint(phonemizer_checkpoint)

    phoneme_symbols = [
        # IPA symbols
        "a",
        "b",
        "d",
        "e",
        "f",
        "g",
        "h",
        "i",
        "j",
        "k",
        "l",
        "m",
        "n",
        "o",
        "p",
        "r",
        "s",
        "t",
        "u",
        "v",
        "w",
        "x",
        "y",
        "z",
        "æ",
        "ç",
        "ð",
        "ø",
        "ŋ",
        "œ",
        "ɐ",
        "ɑ",
        "ɔ",
        "ə",
        "ɛ",
        "ɝ",
        "ɹ",
        "ɡ",
        "ɪ",
        "ʁ",
        "ʃ",
        "ʊ",
        "ʌ",
        "ʏ",
        "ʒ",
        "ʔ",
        "ˈ",
        "ˌ",
        "ː",
        "̃",
        "̍",
        "̥",
        "̩",
        "̯",
        "͡",
        "θ",
        # Punctuation
        "!",
        "?",
        ",",
        ".",
        "-",
        ":",
        ";",
        '"',
        "'",
        "(",
        ")",
        " ",
    ]

    self.tokenizer = SequenceTokenizer(
        phoneme_symbols,
        languages=["de", "en_us"],
        lowercase=True,
        char_repeats=1,
        append_start_end=True,
    )