haoneng.lhn
2023-09-13 5f088a67cd1b18a8260746971f32a6569e0cf2c6
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":