Upload folder using huggingface_hub
Browse files- 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 |
|