funasr/build_utils/build_model_from_file.py
@@ -70,6 +70,8 @@ model.load_state_dict(model_dict) else: model_dict = torch.load(model_file, map_location=device) if task_name == "ss": model_dict = model_dict['model'] if task_name == "diar" and mode == "sond": model_dict = fileter_model_dict(model_dict, model.state_dict()) if task_name == "vad":
