{ "architectures": [ "TinyTransformerForCausalLM" ], "attention_probs_dropout_prob": 0.1, "auto_map": { "AutoConfig": "configuration_tiny_transformer.TinyTransformerConfig", "AutoModelForCausalLM": "modeling_tiny_transformer.TinyTransformerForCausalLM" }, "bos_token_id": 2, "context": 256, "dim": 512, "dropout": 0.1, "dtype": "float32", "eos_token_id": 3, "ff_dim": 2048, "heads": 8, "hidden_dropout_prob": 0.1, "hidden_size": 512, "intermediate_size": 2048, "layers": 8, "max_position_embeddings": 256, "model_type": "tiny_transformer", "num_attention_heads": 8, "num_hidden_layers": 8, "pad_token_id": 0, "tie_word_embeddings": false, "transformers_version": "5.16.1", "unk_token_id": 1, "use_cache": false, "vocab_size": 4096 }