File size: 456 Bytes
0a6452f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import torch
from your_model_file import ChordiaModelClass # 导入你的模型定义类
# 1. 实例化模型结构
model = ChordiaModelClass()
# 2. 加载 .pth 权重 (灵魂归位)
state_dict = torch.load("chordia_v0.0.1-alpha.pth", map_id='cpu')
model.load_state_dict(state_dict)
# 3. 切换到评估模式
model.eval()
# 4. 保存为 Hugging Face 标准格式 (生成 pytorch_model.bin 和 config.json)
model.save_pretrained("./chordia_model_hf") |