{
  "architectures": [
    "Kolibri1ForCausalLM"
  ],
  "model_type": "kolibri1",
  "hidden_size": 2560,
  "num_hidden_layers": 50,
  "num_attention_heads": 48,
  "num_key_value_heads": 4,
  "head_dim": 128,
  "hidden_act": "silu",
  "max_position_embeddings": 262144,
  "rms_norm_eps": 1e-06,
  "vocab_size": 128000,
  "rope_theta": 10000.0,
  "num_experts": 384,
  "num_experts_per_tok": 6,
  "moe_intermediate_size": 512,
  "shared_expert_intermediate_size": 512,
  "norm_topk_prob": false,
  "attention_bias": false,
  "attention_dropout": 0.0,
  "tie_word_embeddings": false,
  "use_cache": true,
  "use_sliding_window": true,
  "sliding_window": 513,
  "layer_types": [
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention"
  ],
  "bos_token_id": null,
  "eos_token_id": 127906,
  "pad_token_id": 127901,
  "dtype": "bfloat16",
  "head_dtype": "float32",
  "quantization_config": {
    "quant_method": "fp8",
    "activation_scheme": "dynamic",
    "weight_block_size": [
      128,
      128
    ],
    "modules_to_not_convert": [
      "model.layers.0.mlp.gate",
      "model.layers.1.mlp.gate",
      "model.layers.2.mlp.gate",
      "model.layers.3.mlp.gate",
      "model.layers.4.mlp.gate",
      "model.layers.5.mlp.gate",
      "model.layers.6.mlp.gate",
      "model.layers.7.mlp.gate",
      "model.layers.8.mlp.gate",
      "model.layers.9.mlp.gate",
      "model.layers.10.mlp.gate",
      "model.layers.11.mlp.gate",
      "model.layers.12.mlp.gate",
      "model.layers.13.mlp.gate",
      "model.layers.14.mlp.gate",
      "model.layers.15.mlp.gate",
      "model.layers.16.mlp.gate",
      "model.layers.17.mlp.gate",
      "model.layers.18.mlp.gate",
      "model.layers.19.mlp.gate",
      "model.layers.20.mlp.gate",
      "model.layers.21.mlp.gate",
      "model.layers.22.mlp.gate",
      "model.layers.23.mlp.gate",
      "model.layers.24.mlp.gate",
      "model.layers.25.mlp.gate",
      "model.layers.26.mlp.gate",
      "model.layers.27.mlp.gate",
      "model.layers.28.mlp.gate",
      "model.layers.29.mlp.gate",
      "model.layers.30.mlp.gate",
      "model.layers.31.mlp.gate",
      "model.layers.32.mlp.gate",
      "model.layers.33.mlp.gate",
      "model.layers.34.mlp.gate",
      "model.layers.35.mlp.gate",
      "model.layers.36.mlp.gate",
      "model.layers.37.mlp.gate",
      "model.layers.38.mlp.gate",
      "model.layers.39.mlp.gate",
      "model.layers.40.mlp.gate",
      "model.layers.41.mlp.gate",
      "model.layers.42.mlp.gate",
      "model.layers.43.mlp.gate",
      "model.layers.44.mlp.gate",
      "model.layers.45.mlp.gate",
      "model.layers.46.mlp.gate",
      "model.layers.47.mlp.gate",
      "model.layers.48.mlp.gate",
      "model.layers.49.mlp.gate"
    ]
  }
}