Yabin Li
2023-09-25 30d8d7a5d3ba4efd6e578db3bce321f7926ec516
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":