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 ######