Support resizing token embeddings

modeling_bert_hash.py

@@ -154,6 +154,12 @@ class BertHashModel(BertPreTrainedModel):
         # Initialize weights and apply final processing
         self.post_init()
 
+    def get_input_embeddings(self):
+        return self.embeddings.word_embeddings.embeddings
+
+    def set_input_embeddings(self, value):
+        self.embeddings.word_embeddings.embeddings = value
+
     def _prune_heads(self, heads_to_prune):
         """
         Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer} See base