def METHOD_NAME(self, wrapped_model: Module, origin2wrapped_name_map: Dict | None, params: List[Dict]) -> List[Dict]:
    param_groups = deepcopy(params)
    origin2wrapped_name_map = origin2wrapped_name_map if origin2wrapped_name_map else {}
    for param_group in param_groups:
        wrapped_names = [origin2wrapped_name_map.get(name, name) for name in param_group['params']]
        param_group['params'] = [p for name, p in wrapped_model.named_parameters() if name in wrapped_names]
    return param_groups