Update modeling_tiny_gpt.py
Browse files- 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 |
-
|
| 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
|