funasr/bin/diar_inference_launch.py
@@ -2,6 +2,9 @@ # Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved. # MIT License (https://opensource.org/licenses/MIT) import torch torch.set_num_threads(1) import argparse import logging import os @@ -133,7 +136,7 @@ param_dict = { "extract_profile": True, "sv_train_config": "sv.yaml", "sv_model_file": "sv.pth", "sv_model_file": "sv.pb", } if "param_dict" in kwargs and kwargs["param_dict"] is not None: for key in param_dict:
