{ "architectures": [ "DumbMathForCausalLM" ], "auto_map": { "AutoConfig": "configuration_dumbmath.DumbMathConfig", "AutoModelForCausalLM": "modeling_dumbmath.DumbMathForCausalLM" }, "model_type": "dumbmath", "vocab_size": 28, "d_model": 128, "n_heads": 4, "n_layers": 4, "d_ff": 252, "max_len": 64, "bos_token_id": 25, "eos_token_id": 26, "pad_token_id": 27 }