fix: bugfix

3 years ago · 5c8e66600e
parent 0674d9ec24
commit 5c8e66600e
1 changed files with 2 additions and 2 deletions
--- a/deepspeech/io/collator.py
+++ b/deepspeech/io/collator.py
@ -354,7 +354,7 @@ class TripletSpeechCollator(SpeechCollator):

            tokens = [[], []]
            for idx, text in enumerate([translation, transcription]):
-                tokens[idx] = tokenids(text, self.keep_transcription_text)
+                tokens[idx] = _tokenids(text, self.keep_transcription_text)

            translation_text.append(tokens[0])
            translation_text_lens.append(tokens[0].shape[0])