cpt core 4

scripts/convert_pth_to_safetensors.py

@@ -4,8 +4,8 @@ import torch
 from safetensors.torch import save_file
 
 
-checkpoint_dir = '../out/pretrain-core-3/hf'
-output_dir = '../out/pretrain-core-3/hf'
+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'))