From f31d12815943e69614a60e76c42006c04b94318d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Th=C3=A9rence?=
 <13496987+Royalphax@users.noreply.github.com>
Date: Sun, 17 Mar 2024 21:09:43 +0100
Subject: [PATCH] Add support for CamembertModel architecture

Adding support for CamembertModel architecture used by :
https://huggingface.co/dangvantuan/sentence-camembert-large
---
 convert-hf-to-gguf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
index cf1f98d66..1e49d56c1 100755
--- a/convert-hf-to-gguf.py
+++ b/convert-hf-to-gguf.py
@@ -1634,7 +1634,7 @@ in chat mode so that the conversation can end normally.")
                 self.post_write_tensors(tensor_map, name, data_torch)
 
 
-@Model.register("BertModel")
+@Model.register("BertModel", "CamembertModel")
 class BertModel(Model):
     model_arch = gguf.MODEL_ARCH.BERT