Multiplication / to_safetensors.py
xcx0902's picture
Upload folder using huggingface_hub
893cb78 verified
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)