config.json

{
    "hidden_size": 512,
    "recursion_steps": 16,
    "vocab_size": 50257,
    "model_type": "trm"
}