嘉渊 %!s(int64=2) %!d(string=hai) anos
pai
achega
f22cb3c12f
Modificáronse 1 ficheiros con 6 adicións e 2 borrados
  1. 6 2
      funasr/build_utils/build_lm_model.py

+ 6 - 2
funasr/build_utils/build_lm_model.py

@@ -34,12 +34,16 @@ class_choices_list = [
 
 def build_lm_model(args):
     # token_list
-    if args.token_list is not None:
-        with open(args.token_list) as f:
+    if isinstance(args.token_list, str):
+        with open(args.token_list, encoding="utf-8") as f:
             token_list = [line.rstrip() for line in f]
         args.token_list = list(token_list)
         vocab_size = len(token_list)
         logging.info(f"Vocabulary size: {vocab_size}")
+    elif isinstance(args.token_list, (tuple, list)):
+        token_list = list(args.token_list)
+        vocab_size = len(token_list)
+        logging.info(f"Vocabulary size: {vocab_size}")
     else:
         vocab_size = None