adityashisharma commited on
Commit
9eda5ba
·
verified ·
1 Parent(s): 348cbf2

Create script/merge_tokenizer_and_ckpt.py

Browse files
Files changed (1) hide show
  1. script/merge_tokenizer_and_ckpt.py +7 -0
script/merge_tokenizer_and_ckpt.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ import shutil, os, pathlib
2
+ dst = pathlib.Path("release")
3
+ dst.mkdir(exist_ok=True)
4
+ shutil.copy("out/tokenizer.json", dst/"tokenizer.json")
5
+ shutil.copy("out/pretrain/gpt_config.json", dst/"gpt_config.json")
6
+ shutil.copy("out/sft/model_sft.pt", dst/"model_sft.pt")
7
+ print("release/ populated.")