FunASR/funasr/utils/build_model.py
speech_asr 680cdb55bb update
2023-04-19 14:49:36 +08:00

14 lines
322 B
Python

import logging
def build_model(args):
if args.token_list is not None:
with open(args.token_list) 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}")