Skip to content

Commit 9b03719

Browse files
authored
convert : add support for CamembertModel architecture (#6119)
Adding support for CamembertModel architecture used by : https://huggingface.co/dangvantuan/sentence-camembert-large
1 parent 3a6efdd commit 9b03719

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert-hf-to-gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1634,7 +1634,7 @@ def write_tensors(self):
16341634
self.post_write_tensors(tensor_map, name, data_torch)
16351635

16361636

1637-
@Model.register("BertModel")
1637+
@Model.register("BertModel", "CamembertModel")
16381638
class BertModel(Model):
16391639
model_arch = gguf.MODEL_ARCH.BERT
16401640

0 commit comments

Comments
 (0)