[fix] Fix offline mode is not working

GitYCC · Aug 27, 2022 · 6e9aeb1 · 6e9aeb1
1 parent d36bad6
commit 6e9aeb1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/g2pw/api.py b/g2pw/api.py
@@ -76,7 +76,7 @@ def __init__(self, model_dir='G2PWModel/', style='bopomofo', model_source=None,
         self.model_source = model_source if model_source else self.config.model_source
         self.turnoff_tqdm = turnoff_tqdm
 
-        self.tokenizer = BertTokenizer.from_pretrained(self.config.model_source)
+        self.tokenizer = BertTokenizer.from_pretrained(self.model_source)
 
         polyphonic_chars_path = os.path.join(model_dir, 'POLYPHONIC_CHARS.txt')
         monophonic_chars_path = os.path.join(model_dir, 'MONOPHONIC_CHARS.txt')