| 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'}) | |