visual-memory/ConvAI2-Qwen-enhanced-gemma-4-26B-A4B-it

{
  "experiment": {
    "model_name": "google/gemma-4-26B-A4B-it",
    "hf_results_repo": "visual-memory/ConvAI2-Qwen-enhanced-gemma-4-26B-A4B-it",
    "results_jsonl": "results/ConvAI2-Qwen-enhanced-gemma-4-26B-A4B-it.jsonl",
    "hf_dataset": "visual-memory/ConvAI2-With-Ids_1k-no-redundancy",
    "hf_mapping_dataset": "visual-memory/ConvAI2-Mapping_1k-no-redundancy",
    "hf_image_dataset": "visual-memory/ConvAI2-Mapping_1k-Qwen-enhanced-no-redundancy",
    "hf_split": "train",
    "num_samples": 0,
    "batch_size": 16,
    "max_seq_len": 4096,
    "compile": false,
    "clear_hf_model_cache": false,
    "device": "cuda:1",
    "dtype": "bfloat16",
    "strategies": [
      "baseline",
      "see_only_you"
    ],
    "name": "convai2-qwen-enhanced"
  },
  "model_default_config": {
    "transformers_version": "5.16.1",
    "architectures": [
      "Gemma4ForConditionalGeneration"
    ],
    "output_hidden_states": false,
    "return_dict": true,
    "dtype": "bfloat16",
    "chunk_size_feed_forward": 0,
    "is_encoder_decoder": false,
    "id2label": {
      "0": "LABEL_0",
      "1": "LABEL_1"
    },
    "label2id": {
      "LABEL_0": 0,
      "LABEL_1": 1
    },
    "problem_type": null,
    "text_config": {
      "architectures": null,
      "output_hidden_states": false,
      "return_dict": true,
      "dtype": "bfloat16",
      "chunk_size_feed_forward": 0,
      "is_encoder_decoder": false,
      "id2label": {
        "0": "LABEL_0",
        "1": "LABEL_1"
      },
      "label2id": {
        "LABEL_0": 0,
        "LABEL_1": 1
      },
      "problem_type": null,
      "vocab_size": 262144,
      "hidden_size": 2816,
      "intermediate_size": 2112,
      "num_hidden_layers": 30,
      "num_attention_heads": 16,
      "num_key_value_heads": 8,
      "head_dim": 256,
      "hidden_activation": "gelu_pytorch_tanh",
      "max_position_embeddings": 262144,
      "initializer_range": 0.02,
      "rms_norm_eps": 1e-06,
      "use_cache": true,
      "pad_token_id": 0,
      "eos_token_id": 1,
      "bos_token_id": 2,
      "tie_word_embeddings": true,
      "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"
        }
      },
      "attention_bias": false,
      "attention_dropout": 0.0,
      "sliding_window": 1024,
      "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"
      ],
      "final_logit_softcapping": 30.0,
      "use_bidirectional_attention": "vision",
      "vocab_size_per_layer_input": 262144,
      "hidden_size_per_layer_input": 0,
      "attention_k_eq_v": true,
      "num_kv_shared_layers": 0,
      "enable_moe_block": true,
      "use_double_wide_mlp": false,
      "num_experts": 128,
      "top_k_experts": 8,
      "moe_intermediate_size": 704,
      "_name_or_path": "",
      "model_type": "gemma4_text",
      "output_attentions": false,
      "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
        }
      }
    },
    "vision_config": {
      "architectures": null,
      "output_hidden_states": false,
      "return_dict": true,
      "dtype": "bfloat16",
      "chunk_size_feed_forward": 0,
      "is_encoder_decoder": false,
      "id2label": {
        "0": "LABEL_0",
        "1": "LABEL_1"
      },
      "label2id": {
        "LABEL_0": 0,
        "LABEL_1": 1
      },
      "problem_type": null,
      "hidden_size": 1152,
      "intermediate_size": 4304,
      "num_hidden_layers": 27,
      "num_attention_heads": 16,
      "num_key_value_heads": 16,
      "head_dim": 72,
      "hidden_activation": "gelu_pytorch_tanh",
      "rms_norm_eps": 1e-06,
      "max_position_embeddings": 131072,
      "attention_bias": false,
      "attention_dropout": 0.0,
      "rope_parameters": {
        "rope_theta": 100.0,
        "rope_type": "default"
      },
      "pooling_kernel_size": 3,
      "patch_size": 16,
      "position_embedding_size": 10240,
      "use_clipped_linears": false,
      "standardize": true,
      "initializer_range": 0.02,
      "_name_or_path": "",
      "default_output_length": 280,
      "global_head_dim": 72,
      "model_type": "gemma4_vision",
      "output_attentions": false
    },
    "audio_config": null,
    "boi_token_id": 255999,
    "eoi_token_id": 258882,
    "image_token_id": 258880,
    "video_token_id": 258884,
    "boa_token_id": 256000,
    "eoa_token_index": 258883,
    "audio_token_id": 258881,
    "initializer_range": 0.02,
    "tie_word_embeddings": true,
    "_name_or_path": "google/gemma-4-26B-A4B-it",
    "eoa_token_id": 258883,
    "eos_token_id": [
      1,
      106
    ],
    "model_type": "gemma4",
    "vision_soft_tokens_per_image": 280,
    "output_attentions": false
  }
}