From cb80c6550fcf5bbbf8d53e2ee9d9e1bea056dde4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <star_dust_chen@foxmail.com>
Date: Tue, 5 Sep 2023 20:18:15 +0800
Subject: [PATCH] Update bert_gen.py

---
 bert_gen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bert_gen.py b/bert_gen.py
index 42dd5ad..c3a3234 100644
--- 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