File size: 382 Bytes
11a0971 |
1 2 3 4 5 6 7 8 9 10 11 |
import safetensors
from safetensors.torch import save_file
tensors = dict()
safetensors_path="/Users/matt/Downloads/hfclone/FundusDRGrading-vit_base_patch14_dinov2/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'})
|