import mindspore as ms

for i in range(160, 164):
    file_name = f"model-00{i}-of-000163.safetensors"
    params = ms.load_checkpoint(file_name, format='safetensors')
    new_params = {}
    for ele in params:
        name = ele.replace('layers.61.', 'layers.4.')
        new_params[name] = params[ele]
                                     
    ms.save_checkpoint(new_params, f"new_{file_name}", format='safetensors')