Update config.json

config.json

@@ -1,16 +1,24 @@
 {
-    "model_name": "alpha3D-v1",
-    "architecture": "SimpleMLP",
-    "version": "1.0",
-    "parameters": {
-        "num_points": 1000,
-        "input_dim": 5,
-        "output_dim": 6000,
-        "hidden_layers": [
-            128,
-            512,
-            1024
-        ]
-    },
-    "license": "Apache-2.0"
+  "architectures": [
+    "Alpha3DModel"
+  ],
+  "model_type": "alpha3d",
+  "auto_map": {
+    "AutoConfig": "configuration_alpha3d.Alpha3DConfig",
+    "AutoModel": "modeling_alpha3d.Alpha3DModel"
+  },
+  "num_points": 1000,
+  "hidden_dims": [
+    128,
+    512,
+    1024
+  ],
+  "vocab": {
+    "red": 0,
+    "green": 1,
+    "blue": 2,
+    "cube": 3,
+    "sphere": 4
+  },
+  "transformers_version": "4.35.2"
 }