Create config.json

config.json

@@ -0,0 +1,12 @@
+{
+  "architectures": [
+    "BigramLanguageModel"
+  ],
+  "n_embd": 64,
+  "n_head": 4,
+  "n_layer": 4,
+  "vocab_size": <vocab_size>,  // Replace with the actual size of your vocabulary
+  "block_size": 32,
+  "dropout": 0.1,
+  "model_type": "custom"
+}