From e946cb093a058f99bc14566fad2ef92bb3db4554 Mon Sep 17 00:00:00 2001 From: Joan Martinez Date: Tue, 23 Apr 2024 09:48:40 +0200 Subject: [PATCH] fix: JinaBertForMaskedLM registration --- convert-hf-to-gguf.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 4619c80959608b..596cddaf6e0e88 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2715,7 +2715,7 @@ def write_tensors(self): self.gguf_writer.add_tensor(new_name, data) -@Model.register("JinaBertModel") +@Model.register("JinaBertModel", "JinaBertForMaskedLM") class JinaBertModel(BertModel): model_arch = gguf.MODEL_ARCH.JINA_BERT @@ -2742,7 +2742,6 @@ def set_vocab(self, *args, **kwargs): self.gguf_writer.add_add_eos_token(True) -JinaBertForMaskedML = JinaBertModel ###### CONVERSION LOGIC ######