Skip to content

Commit d7d6a4e

Browse files
author
Joan Martinez
committed
fix: JinaBertForMaskedLM registration
1 parent 71ff763 commit d7d6a4e

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

convert-hf-to-gguf.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2715,7 +2715,7 @@ def write_tensors(self):
27152715
self.gguf_writer.add_tensor(new_name, data)
27162716

27172717

2718-
@Model.register("JinaBertModel")
2718+
@Model.register("JinaBertModel", "JinaBertForMaskedLM")
27192719
class JinaBertModel(BertModel):
27202720
model_arch = gguf.MODEL_ARCH.JINA_BERT
27212721

@@ -2742,8 +2742,6 @@ def set_vocab(self, *args, **kwargs):
27422742
self.gguf_writer.add_add_eos_token(True)
27432743

27442744

2745-
JinaBertForMaskedML = JinaBertModel
2746-
27472745
###### CONVERSION LOGIC ######
27482746

27492747

0 commit comments

Comments
 (0)