File size: 281 Bytes
0e08ab8 2b6c108 0e08ab8 1c98998 0e08ab8 2b6c108 |
1 2 3 4 5 6 7 8 9 10 11 12 |
import os
import torch
from safetensors.torch import save_file
checkpoint_dir = '../out/cpt-core-pre-4'
output_dir = '../out/cpt-core-pre-4'
state_dict = torch.load(os.path.join(checkpoint_dir, 'model.pth'))
save_file(state_dict, os.path.join(output_dir, 'model.safetensors'))
|