NeuML
/

bert-hash-pico-embeddings

Sentence Similarity

sentence-transformers

feature-extraction

Model card Files Files and versions

davidmezzetti commited on Jun 15

Commit

b2bd0ac

·

verified ·

1 Parent(s): 0083b5a

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",