This commit is contained in:
root
2025-07-29 08:40:51 +00:00
parent d1c354eac7
commit 62d082634e
7 changed files with 71 additions and 68 deletions

View File

@@ -21,6 +21,7 @@ import torch
from safetensors import safe_open
from transformers import AutoTokenizer
def get_args():
parser = ArgumentParser()
@@ -39,6 +40,7 @@ def get_args():
args = parser.parse_args()
return args
if __name__ == "__main__":
args = get_args()
@@ -67,4 +69,3 @@ if __name__ == "__main__":
hf_tensors["llm.model.lm_head.weight"] = hf_tensors["llm.model.model.embed_tokens.weight"]
torch.save(hf_tensors, args.output_path)