Chordia / build.py
Corolin's picture
first commit
0a6452f
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")