aky15
2023-10-20 ab653d3871f72f7f6cd1ac3126b3df722f4c7943
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":