convert : add support for CamembertModel architecture (#6119)

Adding support for CamembertModel architecture used by : https://huggingface.co/dangvantuan/sentence-camembert-large
2024-03-18 09:17:00 +01:00 · 2024-03-18 09:17:00 +01:00 · 9b03719ad7
parent 3a6efdd03c
commit 9b03719ad7
1 changed files with 1 additions and 1 deletions
--- 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