{
  "architectures": [
    "DFlash2DraftModel"
  ],

  "aux_hidden_state_layer_ids": [
    3,
    12,
    21,
    30,
    39
  ],

  "block_size": 7,
  "sample_from_anchor": true,

  "draft_vocab_size": 248320,
  "mask_token_id": 248077,

  "conv_kernel_size": 2,
  "conv_group_size": 16,

  "selector_rank": 256,
  "selector_top_k": 16,

  "draft_ffn_type": "moe",
  "num_experts": 256,
  "num_experts_per_tok": 8,
  "moe_intermediate_size": 512,
  "shared_expert_intermediate_size": 512,
  "norm_topk_prob": true,

  "sliding_window_non_causal": true,

  "dtype": "bfloat16",
  "tie_word_embeddings": false,

  "speculators_model_type": "dflash2",

  "speculators_config": {
    "algorithm": "dflash2",
    "default_proposal_method": "greedy",
    "proposal_methods": [
      {
        "accept_tolerance": 0.0,
        "proposal_type": "greedy",
        "speculative_tokens": 7,
        "verifier_accept_k": 1
      }
    ],
    "verifier": {
      "architectures": [
        "Qwen3_5MoeForConditionalGeneration"
      ],
      "name_or_path": "Kwaipilot/KAT-Coder-V2.5-Dev"
    }
  },

  "transformer_layer_config": {
    "model_type": "qwen3",

    "vocab_size": 248320,
    "hidden_size": 2048,
    "intermediate_size": 512,
    "num_hidden_layers": 3,

    "num_attention_heads": 32,
    "num_key_value_heads": 4,
    "head_dim": 128,

    "hidden_act": "silu",
    "rms_norm_eps": 1e-06,
    "initializer_range": 0.02,

    "attention_bias": false,
    "attention_dropout": 0.0,

    "use_sliding_window": true,
    "sliding_window": 2048,
    "max_window_layers": 0,

    "layer_types": [
      "sliding_attention",
      "sliding_attention",
      "sliding_attention"
    ],

    "max_position_embeddings": 262144,

    "rope_parameters": {
      "rope_theta": 10000000.0,
      "rope_type": "default"
    },

    "use_cache": true,
    "tie_word_embeddings": false,

    "bos_token_id": null,
    "eos_token_id": null,
    "pad_token_id": null
  }
}