From d7d6a4ed466211019366b2b3bcc5efb3edb3c287 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 | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 4619c8095..a132f9c5b 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2715,7 +2715,7 @@ class OlmoModel(Model): 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,8 +2742,6 @@ class JinaBertModel(BertModel): self.gguf_writer.add_add_eos_token(True) -JinaBertForMaskedML = JinaBertModel - ###### CONVERSION LOGIC ######