SauerkrautLM-ColLFM2-450M-v0.1 / sentence_bert_config.json
DavidGF's picture
Integrate with Sentence Transformers via MultiVectorEncoder (#1)
cf36e48
Raw
History Blame Contribute Delete
2.92 kB
{
"transformer_task": "feature-extraction",
"modality_config": {
"text": {
"method": "forward",
"method_output_name": "last_hidden_state"
},
"image": {
"method": "forward",
"method_output_name": "last_hidden_state"
},
"message": {
"method": "forward",
"method_output_name": "last_hidden_state",
"format": "structured"
}
},
"module_output_name": "token_embeddings",
"processing_kwargs": {
"chat_template": {
"chat_template": "sentence_transformers"
}
},
"config_kwargs": {
"do_image_splitting": true,
"downsample_factor": 2,
"encoder_patch_size": 16,
"image_token_id": 396,
"image_token_index": 396,
"max_image_tokens": 256,
"max_num_patches": 1024,
"max_pixels_tolerance": 2.0,
"max_tiles": 10,
"min_image_tokens": 64,
"min_tiles": 2,
"model_type": "lfm2_vl",
"projector_bias": true,
"projector_hidden_act": "gelu",
"projector_hidden_size": 2560,
"text_config": {
"architectures": [
"Lfm2ForCausalLM"
],
"block_auto_adjust_ff_dim": true,
"block_dim": 1024,
"block_ff_dim": 6656,
"block_ffn_dim_multiplier": 1.0,
"block_mlp_init_scale": 1.0,
"block_multiple_of": 256,
"block_norm_eps": 1e-05,
"block_out_init_scale": 1.0,
"block_use_swiglu": true,
"block_use_xavier_init": true,
"conv_L_cache": 3,
"conv_bias": false,
"conv_dim": 1024,
"conv_dim_out": 1024,
"conv_use_xavier_init": true,
"eos_token_id": 7,
"hidden_size": 1024,
"initializer_range": 0.02,
"intermediate_size": 6656,
"layer_types": [
"conv",
"conv",
"full_attention",
"conv",
"conv",
"full_attention",
"conv",
"conv",
"full_attention",
"conv",
"full_attention",
"conv",
"full_attention",
"conv",
"full_attention",
"conv"
],
"max_position_embeddings": 128000,
"model_type": "lfm2",
"norm_eps": 1e-05,
"num_attention_heads": 16,
"num_heads": 16,
"num_hidden_layers": 16,
"num_key_value_heads": 8,
"rope_theta": 1000000.0,
"use_cache": true,
"use_pos_enc": true,
"vocab_size": 65536
},
"tile_size": 512,
"use_image_special_tokens": true,
"use_thumbnail": false,
"vision_config": {
"attention_dropout": 0.0,
"hidden_act": "gelu_pytorch_tanh",
"hidden_size": 768,
"intermediate_size": 3072,
"layer_norm_eps": 1e-06,
"model_type": "siglip2_vision_model",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"num_patches": 256,
"patch_size": 16,
"vision_use_head": false
}
},
"model_kwargs": {
"key_mapping": {
"^model\\.": ""
}
}
}