import safetensors
from safetensors.torch import save_file

tensors = dict()
safetensors_path="/Users/matt/Downloads/hfclone/FundusDRGrading-convnext_small/model.safetensors"
with safetensors.safe_open(safetensors_path, framework="pt") as f:
    for key in f.keys():
        tensors[key] = f.get_tensor(key)

save_file(tensors, safetensors_path, metadata={'format': 'pt'})