| { | |
| "_name_or_path": "./checkpoints/llavaopenchat-7b-other", | |
| "architectures": [ | |
| "MoELLaVAMistralForCausalLM" | |
| ], | |
| "bos_token_id": 1, | |
| "eos_token_id": 32000, | |
| "freeze_mm_mlp_adapter": false, | |
| "hidden_act": "silu", | |
| "hidden_size": 4096, | |
| "image_aspect_ratio": "pad", | |
| "image_projector_type": "mlp2x_gelu", | |
| "initializer_range": 0.02, | |
| "intermediate_size": 14336, | |
| "max_position_embeddings": 8192, | |
| "mm_hidden_size": 1024, | |
| "mm_image_tower": "openai/clip-vit-large-patch14-336", | |
| "mm_projector_lr": null, | |
| "mm_use_im_patch_token": false, | |
| "mm_use_im_start_end": false, | |
| "mm_video_tower": null, | |
| "mm_vision_select_feature": "patch", | |
| "mm_vision_select_layer": -2, | |
| "model_type": "moe_llava_mistral", | |
| "moe": { | |
| "capacity_factor": 1.0, | |
| "ep_size": 1, | |
| "eval_capacity_factor": 2.0, | |
| "min_capacity": 0, | |
| "moe_enable": true, | |
| "moe_layers_idx": [ | |
| 0, | |
| 2, | |
| 4, | |
| 6, | |
| 8, | |
| 10, | |
| 12, | |
| 14, | |
| 16, | |
| 18, | |
| 20, | |
| 22, | |
| 24, | |
| 26, | |
| 28, | |
| 30 | |
| ], | |
| "moe_mode": "sparse", | |
| "num_experts": [ | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4, | |
| 4 | |
| ], | |
| "router_aux_loss_coef": 0.01, | |
| "top_k_experts": 2, | |
| "train_modules": [ | |
| "up_proj", | |
| "down_proj", | |
| "gate_proj", | |
| "wg" | |
| ], | |
| "use_residual": false | |
| }, | |
| "num_attention_heads": 32, | |
| "num_hidden_layers": 32, | |
| "num_key_value_heads": 8, | |
| "pad_token_id": 0, | |
| "rms_norm_eps": 1e-05, | |
| "rope_theta": 10000.0, | |
| "sliding_window": 4096, | |
| "tie_word_embeddings": false, | |
| "tokenizer_padding_side": "right", | |
| "torch_dtype": "bfloat16", | |
| "transformers_version": "4.34.0", | |
| "tune_mm_mlp_adapter": false, | |
| "use_cache": true, | |
| "use_mm_proj": true, | |
| "video_global_proj": false, | |
| "video_projector_type": "linear", | |
| "video_spatial_proj": false, | |
| "video_temproal_proj": false, | |
| "vocab_size": 32002 | |
| } | |