嘉渊 2 лет назад
Родитель
Сommit
69cb9ed756
1 измененных файлов с 6 добавлено и 2 удалено
  1. 6 2
      funasr/build_utils/build_asr_model.py

+ 6 - 2
funasr/build_utils/build_asr_model.py

@@ -227,12 +227,16 @@ class_choices_list = [
 
 def build_asr_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