Update bert_gen.py

2023-09-05 20:18:15 +08:00
parent 908c0bca2c
commit cb80c6550f
1 changed files with 1 additions and 1 deletions
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -6,7 +6,7 @@ from tqdm import tqdm
 from loguru import logger
 from text import cleaned_text_to_sequence, get_bert
 import argparse
-
+import torch.multiprocessing as mp

 def process_line(line):
    rank = mp.current_process()._identity