fix: bugfix

4 years ago · bc29de65fe
parent edb1b185d0
commit bc29de65fe
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])