Skip to content

Commit 71ff763

Browse files
author
Joan Martinez
committed
feat: set proper vocab settings
1 parent 64cd4b1 commit 71ff763

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

convert-hf-to-gguf.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2736,6 +2736,11 @@ def get_tensors(self):
27362736

27372737
yield name, data
27382738

2739+
def set_vocab(self, *args, **kwargs):
2740+
super().set_vocab()
2741+
self.gguf_writer.add_add_bos_token(True)
2742+
self.gguf_writer.add_add_eos_token(True)
2743+
27392744

27402745
JinaBertForMaskedML = JinaBertModel
27412746

0 commit comments

Comments
 (0)