vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: JinaBertForMaskedLM registration

Browse source
This commit is contained in:
Joan Martinez 2024-04-23 09:48:40 +02:00
parent 71ff763e0e
commit d7d6a4ed46
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
convert-hf-to-gguf.py
View file

@ -2715,7 +2715,7 @@ class OlmoModel(Model):
self.gguf_writer.add_tensor(new_name, data) self.gguf_writer.add_tensor(new_name, data)
@Model.register("JinaBertModel") @Model.register("JinaBertModel", "JinaBertForMaskedLM")
class JinaBertModel(BertModel): class JinaBertModel(BertModel):
model_arch = gguf.MODEL_ARCH.JINA_BERT model_arch = gguf.MODEL_ARCH.JINA_BERT
@ -2742,8 +2742,6 @@ class JinaBertModel(BertModel):
self.gguf_writer.add_add_eos_token(True) self.gguf_writer.add_add_eos_token(True)
JinaBertForMaskedML = JinaBertModel
###### CONVERSION LOGIC ###### ###### CONVERSION LOGIC ######