Create script/merge_tokenizer_and_ckpt.py
Browse files
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.")
|