QuantaSparkLabs commited on
Commit
a427778
·
verified ·
1 Parent(s): 9047576

Update modeling_tiny_gpt.py

Browse files
Files changed (1) hide show
  1. modeling_tiny_gpt.py +1 -1
modeling_tiny_gpt.py CHANGED
@@ -193,7 +193,7 @@ class TinyGPTModel(TinyGPTPreTrainedModel):
193
  self.ln_f = nn.LayerNorm(config.n_embd)
194
  self.head = nn.Linear(config.n_embd, config.vocab_size, bias=False)
195
  self.post_init()
196
- # tie_weights will be called by post_init, but we provide the override below.
197
 
198
  def get_input_embeddings(self):
199
  return self.tok_emb
 
193
  self.ln_f = nn.LayerNorm(config.n_embd)
194
  self.head = nn.Linear(config.n_embd, config.vocab_size, bias=False)
195
  self.post_init()
196
+
197
 
198
  def get_input_embeddings(self):
199
  return self.tok_emb