visual-memory/ConvAI2-FLUX-enhanced-gemma-4-E4B-it
Visual Memory Results: convai2-flux-enhanced This dataset contains the scored output of a visual-memory perplexity experiment. Experiment metadata { "experiment": { "model_name": "google/gemma-4-E4B-it", "hf_results_repo": "visual-memory/ConvAI2-FLUX-enhanced-gemma-4-E4B-it", "results_jsonl": "results/ConvAI2-FLUX-enhanced-gemma-4-E4B-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-FLUX-enhanced-gemma-4-E4B-it.
050
Visual Memory Results: convai2-flux-enhanced
This dataset contains the scored output of a visual-memory perplexity experiment.
Experiment metadata
{
"experiment": {
"model_name": "google/gemma-4-E4B-it",
"hf_results_repo": "visual-memory/ConvAI2-FLUX-enhanced-gemma-4-E4B-it",
"results_jsonl": "results/ConvAI2-FLUX-enhanced-gemma-4-E4B-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-FLUX-enhanced-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-flux-enhanced"
},
"model_default_config": {
"transformers_version": "5.17.0",
"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": 2560,
"intermediate_size": 10240,
"num_hidden_layers": 42,
"num_attention_heads": 8,
"num_key_value_heads": 2,
"head_dim": 256,
"hidden_activation": "gelu_pytorch_tanh",
"max_position_embeddings": 131072,
"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": 512,
"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"
],
"final_logit_softcapping": 30.0,
"use_bidirectional_attention": null,
"vocab_size_per_layer_input": 262144,
"hidden_size_per_layer_input": 256,
"attention_k_eq_v": false,
"num_kv_shared_layers": 18,
"enable_moe_block": false,
"use_double_wide_mlp": false,
"num_experts": null,
"top_k_experts": null,
"moe_intermediate_size": null,
"_name_or_path": "",
"expert_intermediate_size": null,
"model_type": "gemma4_text",
"output_attentions": false,
"per_layer_config": {
"05": {
"head_dim": 512
},
"11": {
"head_dim": 512
},
"17": {
"head_dim": 512
},
"23": {
"head_dim": 512
},
"29": {
"head_dim": 512
},
"35": {
"head_dim": 512
},
"41": {
"head_dim": 512
}
}
},
"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": 768,
"intermediate_size": 3072,
"num_hidden_layers": 16,
"num_attention_heads": 12,
"num_key_value_heads": 12,
"head_dim": 64,
"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": "axial"
},
"pooling_kernel_size": 3,
"patch_size": 16,
"position_embedding_size": 10240,
"use_clipped_linears": true,
"standardize": false,
"initializer_range": 0.02,
"_name_or_path": "",
"default_output_length": 280,
"global_head_dim": 64,
"model_type": "gemma4_vision",
"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,
"hidden_size": 1024,
"num_hidden_layers": 12,
"num_attention_heads": 8,
"hidden_act": "silu",
"subsampling_conv_channels": [
128,
32
],
"conv_kernel_size": 5,
"residual_weight": 0.5,
"attention_chunk_size": 12,
"attention_context_left": 13,
"attention_context_right": 0,
"attention_logit_cap": 50.0,
"attention_invalid_logits_value": -1000000000.0,
"use_clipped_linears": true,
"rms_norm_eps": 1e-06,
"gradient_clipping": 10000000000.0,
"output_proj_dims": 1536,
"initializer_range": 0.02,
"_name_or_path": "",
"model_type": "gemma4_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-E4B-it",
"eoa_token_id": 258883,
"eos_token_id": [
1,
106
],
"model_type": "gemma4",
"vision_soft_tokens_per_image": 280,
"output_attentions": false
},
"preprocessing": {
"chunk_size": 32,
"requested_device": "cuda",
"resolved_device": "cuda:0",
"backend": "torchvision",
"used_gpu": true,
"fallback_reason": null
}
}