CoolFace
Datasetpublic

visual-memory/ConvAI2-ERNIE-original-gemma-4-12B-it

Visual Memory Results: convai2-ernie-original This dataset contains the scored output of a visual-memory perplexity experiment. Experiment metadata { "experiment": { "model_name": "google/gemma-4-12B-it", "hf_results_repo": "visual-memory/ConvAI2-ERNIE-original-gemma-4-12B-it", "results_jsonl": "results/ConvAI2-ERNIE-original-gemma-4-12B-it.jsonl", "hf_dataset": "visual-memory/ConvAI2-With-Ids_1k-no-redundancy", "hf_mapping_dataset":… See the full description on the dataset page: https://huggingface.co/datasets/visual-memory/ConvAI2-ERNIE-original-gemma-4-12B-it.

sourceHugging Faceupdated 7d agoView on Hugging Face
0likes50downloads
Dataset Card

Visual Memory Results: convai2-ernie-original

This dataset contains the scored output of a visual-memory perplexity experiment.

Experiment metadata

json
{
  "experiment": {
    "model_name": "google/gemma-4-12B-it",
    "hf_results_repo": "visual-memory/ConvAI2-ERNIE-original-gemma-4-12B-it",
    "results_jsonl": "results/ConvAI2-ERNIE-original-gemma-4-12B-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-ERNIE-original-no-redundancy",
    "hf_split": "train",
    "num_samples": 0,
    "batch_size": 32,
    "preprocess_chunk_size": 32,
    "max_seq_len": 4096,
    "compile": false,
    "clear_hf_model_cache": false,
    "device": "cuda:0",
    "image_preprocessing_device": "cuda",
    "dtype": "bfloat16",
    "strategies": [
      "baseline",
      "see_only_you"
    ],
    "name": "convai2-ernie-original"
  },
  "model_default_config": {
    "transformers_version": "5.17.0",
    "architectures": [
      "Gemma4UnifiedForConditionalGeneration"
    ],
    "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": 3840,
      "intermediate_size": 15360,
      "num_hidden_layers": 48,
      "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",
        "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",
      "attention_k_eq_v": true,
      "num_kv_shared_layers": 0,
      "use_double_wide_mlp": false,
      "_name_or_path": "",
      "enable_moe_block": false,
      "hidden_size_per_layer_input": 0,
      "model_type": "gemma4_unified_text",
      "moe_intermediate_size": null,
      "num_experts": null,
      "top_k_experts": null,
      "vocab_size_per_layer_input": 262144,
      "output_attentions": false,
      "per_layer_config": {
        "05": {
          "head_dim": 512,
          "num_key_value_heads": 1
        },
        "11": {
          "head_dim": 512,
          "num_key_value_heads": 1
        },
        "17": {
          "head_dim": 512,
          "num_key_value_heads": 1
        },
        "23": {
          "head_dim": 512,
          "num_key_value_heads": 1
        },
        "29": {
          "head_dim": 512,
          "num_key_value_heads": 1
        },
        "35": {
          "head_dim": 512,
          "num_key_value_heads": 1
        },
        "41": {
          "head_dim": 512,
          "num_key_value_heads": 1
        },
        "47": {
          "head_dim": 512,
          "num_key_value_heads": 1
        }
      }
    },
    "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,
      "patch_size": 16,
      "pooling_kernel_size": 3,
      "mm_embed_dim": 3840,
      "mm_posemb_size": 1120,
      "rms_norm_eps": 1e-06,
      "output_proj_dims": 3840,
      "initializer_range": 0.02,
      "_name_or_path": "",
      "model_type": "gemma4_unified_vision",
      "num_soft_tokens": 280,
      "output_attentions": false
    },
    "audio_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,
      "audio_embed_dim": 640,
      "rms_norm_eps": 1e-06,
      "initializer_range": 0.02,
      "_name_or_path": "",
      "model_type": "gemma4_unified_audio",
      "output_attentions": false
    },
    "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-12B-it",
    "eos_token_id": [
      1,
      106
    ],
    "model_type": "gemma4_unified",
    "output_attentions": false
  },
  "preprocessing": {
    "chunk_size": 32,
    "requested_device": "cuda",
    "resolved_device": "cuda:0",
    "backend": "torchvision",
    "used_gpu": true,
    "fallback_reason": null
  }
}