convert : add support for CamembertModel architecture (#6119)
Adding support for CamembertModel architecture used by : https://huggingface.co/dangvantuan/sentence-camembert-large
This commit is contained in:
parent
3a6efdd03c
commit
9b03719ad7
|
@ -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
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue