config.json

{
  "bos_token_id": 50256,
  "cycles": 3,
  "dim": 256,
  "dropout": 0.05,
  "eos_token_id": 50256,
  "hidden_dim": 512,
  "high_dilations": [
    4,
    8,
    16,
    32
  ],
  "high_steps": 2,
  "kernel_size": 5,
  "low_dilations": [
    1,
    2,
    4,
    8
  ],
  "low_steps": 6,
  "max_seq_len": 4096,
  "mid_dilations": [
    2,
    4,
    8,
    16
  ],
  "mid_steps": 3,
  "model_type": "trm_hrm",
  "pad_token_id": 50256,
  "tie_word_embeddings": false,
  "transformers_version": "5.0.0",
  "vocab_size": 50257,
  "architectures": [
    "TRMHRMForCausalLM"
  ],
  "auto_map": {
    "AutoConfig": "configuration_trm_hrm.TRMHRMConfig",
    "AutoModelForCausalLM": "modeling_trm_hrm.TRMHRMForCausalLM"
  }
}