luozhangzichen commited on
Commit
7bdd25a
·
verified ·
1 Parent(s): 043b1aa

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. neon213.py +1 -1
neon213.py CHANGED
@@ -167,7 +167,7 @@ class Neon213(PreTrainedModel):
167
  def set_output_embeddings(self, value):
168
  self.head = value
169
 
170
- def tie_weights(self):
171
  # Hugging Face style weight tying
172
  self.head.weight = self.token_emb.weight
173
 
 
167
  def set_output_embeddings(self, value):
168
  self.head = value
169
 
170
+ def tie_weights(self, **kwargs):
171
  # Hugging Face style weight tying
172
  self.head.weight = self.token_emb.weight
173