update eend_ola

嘉渊

2023-07-06 c46a271415d312de777ffbfe3d735e44e39ff68b

update eend_ola

 funasr/build_utils/build_diar_model.py

@@ -178,6 +178,7 @@

def build_diar_model(args):
    # token_list
    if args.token_list is not None:
    if isinstance(args.token_list, str):
        with open(args.token_list, encoding="utf-8") as f:
            token_list = [line.rstrip() for line in f]
@@ -190,6 +191,9 @@
        raise RuntimeError("token_list must be str or list")
    vocab_size = len(token_list)
    logging.info(f"Vocabulary size: {vocab_size}")
    else:
        token_list = None
        vocab_size = None

    # frontend
    if args.input_size is None: