models / model.py
Ahmedtem's picture
Create model.py
75160e6 verified
Raw History Blame Contribute Delete
304 Bytes
import torch
# ุชุญู…ูŠู„ ุงู„ู…ูˆุฏูŠู„
model = YourModelClass()
model.load_state_dict(torch.load("full_model.pth"))
model.eval()
# ุญูุธู‡ ุจุตูŠุบุฉ safetensors (ุงู„ุฃูุถู„ ู„ู€ HF)
from safetensors.torch import save_file
state_dict = model.state_dict()
save_file(state_dict, "model.safetensors")