NeuML
/

bert-hash-nano

Model card Files Files and versions

davidmezzetti commited on Jun 15

Commit

7f9b214

·

verified ·

1 Parent(s): 83c9745

Update modeling_bert_hash.py

Files changed (1) hide show

modeling_bert_hash.py +1 -1

modeling_bert_hash.py CHANGED Viewed

@@ -266,7 +266,7 @@ class BertHashModel(BertPreTrainedModel):
 @auto_docstring
-class BertForMaskedLM(BertPreTrainedModel):
     _tied_weights_keys = {
         "cls.predictions.decoder.weight": "bert.embeddings.word_embeddings.weight",
         "cls.predictions.decoder.bias": "cls.predictions.bias",

 @auto_docstring
+class BertHashForMaskedLM(BertPreTrainedModel):
     _tied_weights_keys = {
         "cls.predictions.decoder.weight": "bert.embeddings.word_embeddings.weight",
         "cls.predictions.decoder.bias": "cls.predictions.bias",