model_config.json

{
  "encoder_dim": 256,
  "hidden_dim": 736,
  "num_heads": 8,
  "ffn_ratio": 4,
  "dropout": 0.05,
  "use_swiglu": true,
  "use_qk_norm": true,
  "zero_init_out_proj": true,
  "gradient_checkpointing": false,
  "prefix_layers": 4,
  "recurrent_layers": 7,
  "recurrent_unrolls": 3,
  "suffix_layers": 4,
  "policy_head_dim": 384,
  "value_hidden": 384,
  "n_value_classes": 3,
  "use_history": false,
  "use_threat_head": false
}