haoneng.lhn
2023-12-08 ace42e5043be992c3a324fc637ba0594f10ad3b6
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":