File size: 225 Bytes
80a1acf
 
 
 
893cb78
 
1
2
3
4
5
6
7
import torch
from safetensors.torch import save_file

model = torch.load("model.pth", weights_only=False)
metadata = {"author": "xcx0902", "year": "2025"}
save_file(model.state_dict(), "model.safetensors", metadata=metadata)