config.json

{
 "architectures": [
 "MLP"
 ],
 "auto_map": {
 "AutoConfig": "configuration_mlp.MLPConfig",
 "AutoModel": "modeling_mlp.MLP"
 },
 "input_size": 784,
 "model_type": "mlp",
 "output_size": 4096,
 "torch_dtype": "float32",
 "transformers_version": "4.44.0"
}