Update modeling_my_gpt.py

modeling_my_gpt.py

@@ -23,10 +23,10 @@ class MyGPTForCausalLM(PreTrainedModel, GenerationMixin):
         # Import your original GPTModel
         self.model = GPTModel({
             "vocab_size": config.vocab_size,
-            "context_length": config.context_length,
-            "emb_dim": config.emb_dim,
-            "n_heads": config.n_heads,
-            "n_layers": config.n_layers,
+            "context_length": config.max_position_embeddings,
+            "emb_dim": config.hidden_size,
+            "n_heads": config.num_attention_heads,
+            "n_layers": config.num_hidden_layers,
             "drop_rate": config.drop_rate,
             "qkv_bias": config.qkv_bias
         })