convert : add support for CamembertModel architecture (ggml-org#6119)

Adding support for CamembertModel architecture used by : https://huggingface.co/dangvantuan/sentence-camembert-large
hodlen · Mar 18, 2024 · 9b03719 · 9b03719
1 parent 3a6efdd
commit 9b03719
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
@@ -1634,7 +1634,7 @@ def write_tensors(self):
                 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