fix: fix linting and editor

This commit is contained in:
Joan Martinez 2024-04-22 17:42:48 +02:00
parent c1c0f4d883
commit 64cd4b1339
2 changed files with 2 additions and 1 deletions

View file

@ -2714,6 +2714,7 @@ class OlmoModel(Model):
self.gguf_writer.add_tensor(new_name, data)
@Model.register("JinaBertModel")
class JinaBertModel(BertModel):
model_arch = gguf.MODEL_ARCH.JINA_BERT