]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
convert : add support for CamembertModel architecture (#6119)
authorThérence <redacted>
Mon, 18 Mar 2024 08:17:00 +0000 (09:17 +0100)
committerGitHub <redacted>
Mon, 18 Mar 2024 08:17:00 +0000 (10:17 +0200)
Adding support for CamembertModel architecture used by :
https://huggingface.co/dangvantuan/sentence-camembert-large

convert-hf-to-gguf.py

index cf1f98d66063385838ea3d8beaac22d5061f8a1b..1e49d56c19514ec9ca171cd8ce5c6a47d2ce1284 100755 (executable)
@@ -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