mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-12-25 02:44:36 +00:00
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)
|
||||
|
||||
|
||||
@Model.register("BertModel")
|
||||
@Model.register("BertModel", "CamembertModel")
|
||||
class BertModel(Model):
|
||||
model_arch = gguf.MODEL_ARCH.BERT
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user