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")