Update bert_gen.py

fishaudio · Sep 30, 2023 · 11dde03 · 11dde03
1 parent d5bea44
commit 11dde03
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bert_gen.py b/bert_gen.py
@@ -8,6 +8,7 @@
 import torch.multiprocessing as mp
 from text.japanese_bert import tokenizer as jp_tokenizer
 from text.chinese_bert import tokenizer
+from text.japanese import text_normalize
 
 def process_line(line):
     rank = mp.current_process()._identity