{
  "aliases": [
    "A",
    "B",
    "C",
    "D",
    "E",
    "F",
    "G",
    "H",
    "I",
    "J",
    "K",
    "L",
    "M",
    "N",
    "O",
    "P",
    "Q",
    "R",
    "S",
    "T",
    "U",
    "V",
    "W",
    "X",
    "Y",
    "Z",
    "a",
    "b",
    "c",
    "d",
    "e",
    "f",
    "g",
    "h",
    "i",
    "j",
    "k",
    "l",
    "m",
    "n",
    "o",
    "p",
    "q",
    "r",
    "s",
    "t",
    "u",
    "v",
    "w",
    "x",
    "y",
    "z",
    "AA",
    "AB",
    "AC",
    "AD",
    "AE",
    "AF",
    "AG",
    "AH",
    "AI",
    "AJ",
    "AK",
    "AL"
  ],
  "auto_map": {
    "AutoConfig": "configuration_hev.HevConfig",
    "AutoModel": "modeling_hev.HevForDecisions"
  },
  "backbone_config": {
    "_name_or_path": "google/diffusiongemma-26B-A4B-it",
    "architectures": [
      "DiffusionGemmaForBlockDiffusion"
    ],
    "boi_token_id": 255999,
    "canvas_length": 256,
    "chunk_size_feed_forward": 0,
    "dtype": "bfloat16",
    "eoi_token_id": 258882,
    "eos_token_id": [
      1,
      106
    ],
    "id2label": {
      "0": "LABEL_0",
      "1": "LABEL_1"
    },
    "image_token_id": 258880,
    "initializer_range": 0.02,
    "is_encoder_decoder": false,
    "label2id": {
      "LABEL_0": 0,
      "LABEL_1": 1
    },
    "model_type": "diffusion_gemma",
    "output_attentions": false,
    "output_hidden_states": false,
    "problem_type": null,
    "return_dict": true,
    "text_config": {
      "_name_or_path": "",
      "architectures": null,
      "attention_bias": false,
      "attention_dropout": 0.0,
      "bos_token_id": 2,
      "chunk_size_feed_forward": 0,
      "dtype": "bfloat16",
      "eos_token_id": 1,
      "final_logit_softcapping": 30.0,
      "head_dim": 256,
      "hidden_activation": "gelu_pytorch_tanh",
      "hidden_size": 2816,
      "id2label": {
        "0": "LABEL_0",
        "1": "LABEL_1"
      },
      "initializer_range": 0.02,
      "intermediate_size": 2112,
      "is_encoder_decoder": false,
      "label2id": {
        "LABEL_0": 0,
        "LABEL_1": 1
      },
      "layer_types": [
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "full_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "full_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "full_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "full_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "sliding_attention",
        "full_attention"
      ],
      "max_position_embeddings": 262144,
      "model_type": "diffusion_gemma_text",
      "moe_intermediate_size": 704,
      "num_attention_heads": 16,
      "num_experts": 128,
      "num_hidden_layers": 30,
      "num_key_value_heads": 8,
      "output_attentions": false,
      "output_hidden_states": false,
      "pad_token_id": 0,
      "per_layer_config": {
        "05": {
          "head_dim": 512,
          "num_key_value_heads": 2
        },
        "11": {
          "head_dim": 512,
          "num_key_value_heads": 2
        },
        "17": {
          "head_dim": 512,
          "num_key_value_heads": 2
        },
        "23": {
          "head_dim": 512,
          "num_key_value_heads": 2
        },
        "29": {
          "head_dim": 512,
          "num_key_value_heads": 2
        }
      },
      "problem_type": null,
      "return_dict": true,
      "rms_norm_eps": 1e-06,
      "rope_parameters": {
        "full_attention": {
          "partial_rotary_factor": 0.25,
          "rope_theta": 1000000.0,
          "rope_type": "proportional"
        },
        "sliding_attention": {
          "rope_theta": 10000.0,
          "rope_type": "default"
        }
      },
      "sliding_window": 1024,
      "tie_word_embeddings": true,
      "top_k_experts": 8,
      "use_bidirectional_attention": "vision",
      "vocab_size": 262144
    },
    "tie_word_embeddings": true,
    "transformers_version": "5.17.0",
    "vision_config": {
      "_name_or_path": "",
      "architectures": null,
      "attention_bias": false,
      "attention_dropout": 0.0,
      "chunk_size_feed_forward": 0,
      "default_output_length": 280,
      "dtype": "bfloat16",
      "global_head_dim": 72,
      "head_dim": 72,
      "hidden_activation": "gelu_pytorch_tanh",
      "hidden_size": 1152,
      "id2label": {
        "0": "LABEL_0",
        "1": "LABEL_1"
      },
      "initializer_range": 0.02,
      "intermediate_size": 4304,
      "is_encoder_decoder": false,
      "label2id": {
        "LABEL_0": 0,
        "LABEL_1": 1
      },
      "max_position_embeddings": 131072,
      "model_type": "gemma4_vision",
      "num_attention_heads": 16,
      "num_hidden_layers": 27,
      "num_key_value_heads": 16,
      "output_attentions": false,
      "output_hidden_states": false,
      "patch_size": 16,
      "pooling_kernel_size": 3,
      "position_embedding_size": 10240,
      "problem_type": null,
      "return_dict": true,
      "rms_norm_eps": 1e-06,
      "rope_parameters": {
        "rope_theta": 100.0,
        "rope_type": "axial"
      },
      "standardize": true,
      "use_clipped_linears": false
    },
    "vision_soft_tokens_per_image": 280
  },
  "backbone_id": "google/diffusiongemma-26B-A4B-it",
  "canvas_length": 64,
  "decoder_norm": "model.decoder.norm",
  "encoder_norm": "model.encoder.language_model.norm",
  "extra_layers": {
    "decoder": [
      "model.decoder.layers.14",
      "model.decoder.layers.22"
    ],
    "encoder": [
      "model.encoder.language_model.layers.14",
      "model.encoder.language_model.layers.22"
    ]
  },
  "head_config": {
    "chunk_size": 64,
    "d_backbone": 8448,
    "d_conv": 7,
    "d_model": 512,
    "d_state": 64,
    "expand": 2,
    "headdim": 64,
    "mlp_mult": 2,
    "n_blocks": 3,
    "n_types": 3
  },
  "max_prompt_tokens": 1024,
  "member_temperatures": [
    [
      1.338762879371643,
      1.776159405708313,
      1.816263198852539
    ],
    [
      1.4009417295455933,
      1.899003267288208,
      1.8320508003234863
    ],
    [
      1.3898975849151611,
      1.61630380153656,
      1.888205885887146
    ]
  ],
  "model_type": "hev",
  "n_heads": 3,
  "prompt_format_version": 1,
  "seed_mode": "pad",
  "slot_token_id": 0,
  "temperatures": [
    1.338762879371643,
    1.776159405708313,
    1.816263198852539
  ],
  "transformers_version": "5.17.0"
}