Model Versioning

v.py

from safetensors.torch import save_model, load_model
from safetensors import safe_open
import json

config_dict = {'version': '1.0'}
metadata = {k: json.dumps(v) if not isinstance(v, str) else v for k, v in config_dict.items()}

save_model(model, "model.safetensors", metadata=metadata)

# ...

with safe_open("model.safetensors", framework="pt") as f:
    metadata = f.metadata()
    
config_dict = {k: json.loads(v) if not isinstance(v, str) else v for k, v in metadata.items()}
print(f"Ver: {config_dict['version']}")

load_model(model, "model.safetensors")