Quellcode durchsuchen

convert : add support for CamembertModel architecture (#6119)

Adding support for CamembertModel architecture used by :
https://huggingface.co/dangvantuan/sentence-camembert-large
Thérence vor 1 Jahr
Ursprung
Commit
9b03719ad7
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      convert-hf-to-gguf.py

+ 1 - 1
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)
                 self.post_write_tensors(tensor_map, name, data_torch)
 
 
 
 
-@Model.register("BertModel")
+@Model.register("BertModel", "CamembertModel")
 class BertModel(Model):
 class BertModel(Model):
     model_arch = gguf.MODEL_ARCH.BERT
     model_arch = gguf.MODEL_ARCH.BERT