Bug fix on tortoise inference

coqui-ai · Sep 28, 2023 · 01c04f7 · 01c04f7
1 parent bb05dcb
commit 01c04f7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/TTS/tts/layers/tortoise/tokenizer.py b/TTS/tts/layers/tortoise/tokenizer.py
@@ -5,9 +5,12 @@
 
 from TTS.tts.utils.text.cleaners import english_cleaners
 
+DEFAULT_VOCAB_FILE = os.path.join(
+    os.path.dirname(os.path.realpath(__file__)), "../../utils/assets/tortoise/tokenizer.json"
+)
 
 class VoiceBpeTokenizer:
-    def __init__(self, vocab_file=None, vocab_str=None):
+    def __init__(self, vocab_file=DEFAULT_VOCAB_FILE, vocab_str=None):
         self.tokenizer = None
         if vocab_file is not None:
             self.tokenizer = Tokenizer.from_file(vocab_file)