Merge pull request #1 from Royalphax/Royalphax-patch-1

Add support for CamembertModel architecture
This commit is contained in:
Thérence 2024-03-17 21:15:09 +01:00 committed by GitHub
commit 7db32e531d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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