Update config.json

config.json

@@ -2,11 +2,17 @@
   "architectures": [
     "TopAIImageGenerator"
   ],
-  "dtype": "float32",
+  "model_type": "topai_image_gen",
+  "auto_map": {
+    "AutoConfig": "configuration_pixel.TopAIImageConfig",
+    "AutoModel": "modeling_pixel.TopAIImageGenerator"
+  },
+  "input_dim": 768,
   "hidden_dim": 512,
   "image_channels": 3,
-  "input_dim": 768,
-  "model_type": "topai_image_gen",
   "resolution": 128,
-  "transformers_version": "5.2.0"
+  "torch_dtype": "float32",
+  "transformers_version": "5.2.0",
+  "initializer_range": 0.02,
+  "model_name": "Pixel-1"
 }