import json
import torch
from safetensors.torch import save_file

parameters = json.loads(open("parameter.json").read())
model_path = parameters["model_path"]

model = torch.load(model_path, weights_only=False)
save_file(model.state_dict(), model_path.rstrip(".pth") + ".safetensors")