Upload modeling_hebrewgpt.py with huggingface_hub
Browse files- modeling_hebrewgpt.py +1 -1
modeling_hebrewgpt.py
CHANGED
|
@@ -134,7 +134,7 @@ class HebrewGPTPreTrainedModel(PreTrainedModel):
|
|
| 134 |
|
| 135 |
|
| 136 |
class HebrewGPTForCausalLM(HebrewGPTPreTrainedModel):
|
| 137 |
-
_tied_weights_keys =
|
| 138 |
|
| 139 |
def __init__(self, config: HebrewGPTConfig):
|
| 140 |
super().__init__(config)
|
|
|
|
| 134 |
|
| 135 |
|
| 136 |
class HebrewGPTForCausalLM(HebrewGPTPreTrainedModel):
|
| 137 |
+
_tied_weights_keys = {"head.weight": "tok_emb.weight"}
|
| 138 |
|
| 139 |
def __init__(self, config: HebrewGPTConfig):
|
| 140 |
super().__init__(config)
|