{ "model_type": "llama", "architectures": [ "TinyStoriesGPT" ], "auto_map": { "AutoModel": "modeling_tinystories.TinyStoriesGPT", "AutoModelForCausalLM": "modeling_tinystories.TinyStoriesGPT" }, "hidden_size": 512, "intermediate_size": 1408, "num_hidden_layers": 12, "num_attention_heads": 8, "num_key_value_heads": 4, "head_dim": 64, "vocab_size": 8192, "max_position_embeddings": 512, "rope_theta": 10000.0, "tie_word_embeddings": true, "torch_dtype": "float32", "transformers_version": "4.45.0" }