Update config.json

config.json

@@ -1,28 +1,15 @@
 {
-  "architectures": [
-    "DenseTransformerForCausalLM"
-  ],
-  "attention_dropout": 0.1,
-  "attn_eps": 1e-06,
-  "bos_token_id": 1,
-  "d_head": 64,
-  "eos_token_id": 2,
-  "ffn_eps": 1e-06,
-  "hidden_dropout": 0.1,
+  "model_type": "tiny_gpt",
+  "architectures": ["TinyGPTLMHeadModel"],
+  "vocab_size": 50256,
   "hidden_size": 768,
-  "intermediate_size": 3072,
-  "max_position_embeddings": 1024,
-  "model_type": "dense_transformer",
   "num_attention_heads": 12,
   "num_hidden_layers": 5,
-  "num_key_value_heads": 8,
-  "pad_token_id": 0,
-  "rms_norm_eps": 1e-06,
-  "seq_len": 512,
-  "tie_word_embeddings": false,
-  "torch_dtype": "float32",
-  "transformers_version": "4.36.0",
-  "use_cache": true,
-  "vocab_size": 50256,
-  "window_size": 128
+  "intermediate_size": 3072,
+  "max_position_embeddings": 1024,
+  "rms_norm_eps": 1e-6,
+  "rope_theta": 10000.0,
+  "attention_dropout": 0.1,
+  "hidden_dropout": 0.1,
+  "use_cache": true
 }