update repo

嘉渊

2023-06-19 e2f8fa4b64bb7c058513bd1f5c970385d0d84672

update repo

 funasr/build_utils/build_model_from_file.py

@@ -74,6 +74,9 @@
            model_dict = torch.load(model_file, map_location=device)
    if task_name == "diar" and mode == "sond":
        model_dict = fileter_model_dict(model_dict, model.state_dict())
    if task_name == "vad":
        model.encoder.load_state_dict(model_dict)
    else:
    model.load_state_dict(model_dict)
    if model_name_pth is not None and not os.path.exists(model_name_pth):
        torch.save(model_dict, model_name_pth)