ronnengmail commited on
Commit
c0bf8b4
·
verified ·
1 Parent(s): 3140dfa

Upload modeling_hebrewgpt.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 = ["head.weight"]
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)