diff options
author | Thérence <13496987+Royalphax@users.noreply.github.com> | 2024-03-18 09:17:00 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-03-18 10:17:00 +0200 |
commit | 9b03719ad712e2dc36c5c0c20f352bf3e4bda332 (patch) | |
tree | 3c104ad917d8edeceb8e3e474cc358cd1f8451ee | |
parent | 3a6efdd03c46c5ba08e43880d34260c02dd9999b (diff) |
convert : add support for CamembertModel architecture (#6119)
Adding support for CamembertModel architecture used by :
https://huggingface.co/dangvantuan/sentence-camembert-large
-rwxr-xr-x | convert-hf-to-gguf.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index cf1f98d6..1e49d56c 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 |