add config

config.json

@@ -0,0 +1,7 @@
+{
+  "n_layers": 2,
+  "model_type": "my_model",
+  "auto_map": {
+    "AutoConfig": "configuration_my_model.MyModelConfig"
+  }
+}

configuration_my_model.py

@@ -0,0 +1,13 @@
+from transformers import PretrainedConfig
+
+
+class MyModelConfig(PretrainedConfig):
+    model_type = "my_model"
+
+    def __init__(
+        self,
+        n_layers=2,
+        **kwargs,
+    ):
+        self.n_layers = n_layers
+        super().__init__(**kwargs)