WAM-Diff / config.json
cuijh26's picture
Upload folder using huggingface_hub
8da1b65 verified
{
"_name_or_path": "exp/rl_sample_with_threshold_moe_multigspo/checkpoint-2000/",
"add_faster_video": false,
"add_time_instruction": false,
"architectures": [
"LlavaLLaDAModelLM"
],
"attention_bias": false,
"attention_dropout": 0.0,
"auto_map": {
"AutoConfig": "configuration_llada.LLaDAConfig",
"AutoModel": "modeling_llada.LLaDAModelLM",
"AutoModelForCausalLM": "modeling_llada.LLaDAModelLM"
},
"bos_token_id": 128000,
"capacity_factor": 0.1,
"eos_token_id": 126081,
"faster_token_stride": 10,
"force_sample": false,
"hidden_act": "silu",
"hidden_size": 4096,
"image_aspect_ratio": "anyres_max_2",
"image_crop_resolution": null,
"image_grid_pinpoints": [
[
384,
384
],
[
384,
768
],
[
384,
1152
],
[
384,
1536
],
[
384,
1920
],
[
384,
2304
],
[
768,
384
],
[
768,
768
],
[
768,
1152
],
[
768,
1536
],
[
768,
1920
],
[
768,
2304
],
[
1152,
384
],
[
1152,
768
],
[
1152,
1152
],
[
1152,
1536
],
[
1152,
1920
],
[
1152,
2304
],
[
1536,
384
],
[
1536,
768
],
[
1536,
1152
],
[
1536,
1536
],
[
1536,
1920
],
[
1536,
2304
],
[
1920,
384
],
[
1920,
768
],
[
1920,
1152
],
[
1920,
1536
],
[
1920,
1920
],
[
1920,
2304
],
[
2304,
384
],
[
2304,
768
],
[
2304,
1152
],
[
2304,
1536
],
[
2304,
1920
],
[
2304,
2304
]
],
"image_split_resolution": null,
"initializer_range": 0.02,
"intermediate_size": 12288,
"max_position_embeddings": 16384,
"mm_hidden_size": 1152,
"mm_newline_position": "grid",
"mm_patch_merge_type": "spatial_unpad",
"mm_projector_lr": null,
"mm_projector_type": "mlp2x_gelu",
"mm_resampler_type": null,
"mm_spatial_pool_mode": "bilinear",
"mm_spatial_pool_stride": null,
"mm_tunable_parts": "mm_vision_tower,mm_mlp_adapter,mm_language_model",
"mm_use_im_patch_token": false,
"mm_use_im_start_end": false,
"mm_vision_select_feature": "patch",
"mm_vision_select_layer": -2,
"mm_vision_tower": "model/siglip2-so400m-patch14-384",
"mm_vision_tower_lr": 2e-06,
"model_type": "llava_llada",
"moe_choice": "expert",
"moe_lora_alpha": 32,
"moe_lora_dropout": 0.05,
"moe_lora_in_features": 4096,
"moe_lora_out_features": 4096,
"moe_lora_rank": 32,
"moe_router_enable_expert_bias": null,
"moe_router_score_function": null,
"num_attention_heads": 32,
"num_experts": 64,
"num_experts_per_tok": 64,
"num_hidden_layers": 32,
"num_key_value_heads": 32,
"pos_skipping_range": 4096,
"pretraining_tp": 1,
"rms_norm_eps": 1e-05,
"rope_scaling": null,
"rope_theta": 500000.0,
"tie_word_embeddings": false,
"tokenizer_model_max_length": 8192,
"tokenizer_padding_side": "right",
"torch_dtype": "bfloat16",
"transformers_version": "4.44.0",
"use_cache": false,
"use_mm_proj": true,
"use_pos_skipping": false,
"vision_tower_pretrained": null,
"vocab_size": 146350
}