{ "architectures": [ "SimpleNN" ], "auto_map": { "AutoConfig": "configuration_simple_model.SimpleNNConfig", "AutoModel": "simpleModel.SimpleNN" }, "input_size": 784, "model_type": "simple_nn", "num_classes": 10, "torch_dtype": "float32", "transformers_version": "4.44.2" }