Skip to content

Commit 5592c08

Browse files
committed
add bos note
1 parent 9ecf1bd commit 5592c08

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5049,6 +5049,7 @@ def set_vocab(self):
50495049
special_vocab._set_special_token("unk", tokenizer.get_added_vocab()["<|endoftext|>"])
50505050
# exclude glm-edge 1.5B & 4B
50515051
if self.hparams.get("partial_rotary_factor", 1.0) == 0.5:
5052+
# GLM4 model has no bos token set in its tokenizer config, set it manually to [gMASK]
50525053
special_vocab._set_special_token("bos", tokenizer.get_added_vocab()["[gMASK]"])
50535054
special_vocab.add_to_gguf(self.gguf_writer)
50545055

0 commit comments

Comments
 (0)