fix load jieba once

wenet-e2e · Nov 1, 2024 · ab2f6f8 · ab2f6f8
1 parent 5bd0663
commit ab2f6f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/cli/punc_model.py b/wenet/cli/punc_model.py
@@ -16,14 +16,14 @@ def __init__(self, model_dir: str) -> None:
 
         self.model = torch.jit.load(model_path)
         self.tokenizer = CharTokenizer(units_path)
-        # self.model = torch.jit.load(model_path)
         self.device = torch.device("cpu")
         self.use_jieba = False
 
         self.punc_table = ['<unk>', '', '，', '。', '？', '、']
 
     def split_words(self, text: str):
         if not self.use_jieba:
+            self.use_jieba = True
             import logging
 
             import jieba