add config.json

config.json

@@ -0,0 +1,5 @@
+{
+    "hidden_size": 128,
+    "model_type": "custom-model",
+    "architectures": ["CustomModel"]
+  }