tiny-random-molmo2 / config.json
rkazants's picture
Upload tiny random Molmo2 (allenai/MolmoWeb-4B architecture) (#1)
ae23feb
Raw History Blame Contribute Delete
2.52 kB
{
"adapter_config": {
"attention_dropout": 0.0,
"attn_implementation": "sdpa",
"float32_attention": true,
"head_dim": 32,
"hidden_act": "silu",
"hidden_size": 64,
"image_feature_dropout": 0.0,
"initializer_range": 0.02,
"intermediate_size": 128,
"model_type": "molmo2",
"num_attention_heads": 2,
"num_key_value_heads": 2,
"pooling_attention_mask": true,
"residual_dropout": 0.0,
"text_hidden_size": 128,
"vit_layers": [
-1,
-2
]
},
"architectures": [
"Molmo2ForConditionalGeneration"
],
"auto_map": {
"AutoConfig": "configuration_molmo2.Molmo2Config",
"AutoModelForImageTextToText": "modeling_molmo2.Molmo2ForConditionalGeneration"
},
"dtype": "float32",
"frame_end_token_id": 151944,
"frame_start_token_id": 151943,
"image_col_id": 151939,
"image_end_token_id": 151937,
"image_high_res_id": 151938,
"image_low_res_id": 151942,
"image_patch_id": 151938,
"image_start_token_id": 151936,
"initializer_range": 0.02,
"low_res_image_start_token_id": 151940,
"model_type": "molmo2",
"text_config": {
"additional_vocab_size": 128,
"attention_dropout": 0.0,
"attn_implementation": "sdpa",
"embedding_dropout": 0.0,
"head_dim": 32,
"hidden_act": "silu",
"hidden_size": 128,
"initializer_range": 0.02,
"intermediate_size": 256,
"layer_norm_eps": 1e-06,
"max_position_embeddings": 10240,
"model_type": "molmo2_text",
"norm_after": false,
"num_attention_heads": 4,
"num_hidden_layers": 3,
"num_key_value_heads": 2,
"qk_norm_type": "qwen3",
"qkv_bias": false,
"residual_dropout": 0.0,
"rope_scaling": null,
"rope_scaling_layers": null,
"rope_theta": 1000000.0,
"use_cache": true,
"use_qk_norm": true,
"vocab_size": 151936
},
"tie_word_embeddings": false,
"transformers_version": "4.57.6",
"use_cache": true,
"use_frame_special_tokens": false,
"vit_config": {
"attention_dropout": 0.0,
"attn_implementation": "sdpa",
"float32_attention": true,
"head_dim": 32,
"hidden_act": "gelu_pytorch_tanh",
"hidden_size": 64,
"image_default_input_size": [
378,
378
],
"image_num_pos": 729,
"image_patch_size": 14,
"initializer_range": 0.02,
"intermediate_size": 128,
"layer_norm_eps": 1e-06,
"model_type": "molmo2",
"num_attention_heads": 2,
"num_hidden_layers": 3,
"num_key_value_heads": 2,
"residual_dropout": 0.0
}
}