雾聪
2023-12-13 de87e1d180d214e1f49682d2b5fb7c9d2c89ae7e
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":