root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-28 12:24:35 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

convert-gptneox-h5-to-gguf.py : Special tokens

Browse Source
This commit is contained in:
klosax 2023-08-01 14:30:00 +02:00 committed by GitHub
parent da4900e835
commit e7a741695c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 38 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
convert-gptneox-h5-to-gguf.py
View File

@ -58,7 +58,7 @@ for name in list_vars.keys():
gguf_writer = gguf.GGUFWriter.open(fname_out) gguf_writer = gguf.GGUFWriter.open(fname_out)
# This must be changed when adding/deleting kv # This must be changed when adding/deleting kv
kv_count = 14 kv_count = 17
print("tensors " + str(tensor_count) + " kv " + str(kv_count)) print("tensors " + str(tensor_count) + " kv " + str(kv_count))
@ -101,9 +101,43 @@ if Path(dir_model + "/tokenizer.json").is_file():
merges = tokenizer["model"]["merges"] merges = tokenizer["model"]["merges"]
gguf_writer.write_tokenizer_model("gpt2") gguf_writer.write_tokenizer_model("gpt2")
gguf_writer.write_token_list(tokens) gguf_writer.write_token_list(tokens)
gguf_writer.write_token_merges(merges) gguf_writer.write_token_merges(merges)
if "added_tokens" in tokenizer and Path(dir_model + "/tokenizer_config.json").is_file():
print("Adding special token ids")
with open(dir_model + "/tokenizer_config.json", "r", encoding="utf-8") as f:
tokenizer_config = json.load(f)
# find special token ids
if "bos_token" in tokenizer_config:
for key in tokenizer["added_tokens"]:
if key["content"] == tokenizer_config["bos_token"]:
gguf_writer.write_uint32("tokenizer.ggml.bos_token_id", key["id"] )
if "eos_token" in tokenizer_config:
for key in tokenizer["added_tokens"]:
if key["content"] == tokenizer_config["eos_token"]:
gguf_writer.write_uint32("tokenizer.ggml.eos_token_id", key["id"] )
if "unk_token" in tokenizer_config:
for key in tokenizer["added_tokens"]:
if key["content"] == tokenizer_config["unk_token"]:
gguf_writer.write_uint32("tokenizer.ggml.unknown_token_id", key["id"] )
if "sep_token" in tokenizer_config:
for key in tokenizer["added_tokens"]:
if key["content"] == tokenizer_config["sep_token"]:
gguf_writer.write_uint32("tokenizer.ggml.separator_token_id", key["id"] )
if "pad_token" in tokenizer_config:
for key in tokenizer["added_tokens"]:
if key["content"] == tokenizer_config["pad_token"]:
gguf_writer.write_uint32("tokenizer.ggml.padding_token_id", key["id"] )
# TENSORS # TENSORS