Skip to content

Commit 1004327

Browse files
committed
add set_vocab to modernbert convert class
1 parent 31e87e4 commit 1004327

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
@@ -3940,6 +3940,11 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
39403940
class ModernBertModel(BertModel):
39413941
model_arch = gguf.MODEL_ARCH.MODERN_BERT
39423942

3943+
def set_vocab(self):
3944+
self._set_vocab_gpt2()
3945+
self.gguf_writer.add_add_bos_token(True)
3946+
self.gguf_writer.add_add_eos_token(True)
3947+
39433948
def set_gguf_parameters(self):
39443949
super().set_gguf_parameters()
39453950
self._try_set_pooling_type()

0 commit comments

Comments
 (0)